最近GPU 上重新安裝pytorch 出現了一些問題,整理一下正確安裝pytorch
1.打開pytorch 官網 https://pytorch.org/
如圖生成自己安裝環境對應的命令 ,如果找不到對應版本,可以直接替換版本號即可,例如在 cuda 9.0 下安裝 torch
conda install pytorch torchvision cudatoolkit=9.0 -c pytorch -c defaults -c numba/label/dev
測試安裝是否成功 (torch.cuda.is_available())