首先說明
本文並不是安裝教程，網上有很多，這裡只是自己遇到的一些問題
我是以前安裝的tensorflow-gpu的，但是發現現在的學術論文大部分都是用pytorch復現的，因此才去安裝的pytorch-gpu
檢視自己安裝的CUDA
nvcc -V
這裡我提供一個安裝tensorflow時所用的CUDA對應表
安裝cuDNN時版本一定要對應正確
安裝完CUDA時要改一下環境變數
只用新增這個就好，其他三個CUDA會自動為你新增
我的是cuda-11.2
但是官網沒有配套的，直接就挑自己喜歡的下就行Start Locally | PyTorch
這裡強調：千萬不要相信在其他網站上的下載命令。有可能你下載的就是cpu版本
安裝的時候可能會遇到
Solving environment: failed with initial frozen solve. Retrying with flexible solve.
不要擔心，不用管它，耐心等一等，就會出現（y/n）選擇y就可以了
這裡要看一眼你的pytorch是不是cuda版本的，不是要重新下載
測試安裝是否成功
import torch print(torch.__version__) #注意是雙下劃線 print(torch.version.cuda) print(torch.cuda.is_available()) print(torch.cuda.get_device_name())
當然你這些都做完了，還是出現了False
失敗後的操作
開啟你的anaconda
將這裡關於pytorch的資訊全部刪除，再按照前面的流程走一遍，相信你自己一定會出現