manjaro目前中國資料偏少,踩了很多坑。
安裝gpu版本就這么幾個步驟 1 安裝英偉達的驅動cuda 2 安裝 cudnn 3 安裝支持gpu的pytorch 或者其他的運算框架
manjaro中文資料很少,,,一步一步先裝cuda 再裝cudnn 好麻煩,各種錯誤
實際上就這一句就解決了:
1 裝 cuda 和 cudnn
sudo pacman -S python-pytorch-cuda
就這一句就自動吧 cuda cudnn 都裝好了,
這句話執行中間會失敗很多次,,多幾次重新執行,它把所有包下載好了 就能成功了。
然后使用命令查看一下自己的cuda 是什么版本
optirun nvidia-smi
2 安裝支持gpu的pytorch
去官方網站 https://pytorch.org/
找對應自己的pyorch安裝
我的cuda 是10.1 但是我發現網站上pytorch 支持cuda的只到10, 所以我就安裝了cuda10的pytorch。。。。后來也成功了
3 驗證一下
用 optirun python 開python交互,,
import torch
torch.cuda.is_available()
就成功了。
據說manjaro系統不會自動切換獨立顯卡,,所以運行時候要加上optirun 才能調用獨立顯卡