linux manjaro 配置 pytorch gpu 環境

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 才能調用獨立顯卡