說明:torch.cuda.is_available()
這個指令的做用是看你電腦的 GPU 可否被 PyTorch 調用。
若是返回的結果是 False,能夠按照如下過程進行排查。html
Step1:確認硬件支持,確認你的 GPU是否支持 CUDA(是否支持被 PyTorch 調用)python
1.肯定計算機中是不是獨立顯卡,是不是 NVIDIA 顯卡。能夠從 任務管理器 或者 設備管理器 來查看顯卡的型號。性能
有「NVIDIA GeForce 840M」字樣信息表示有獨立顯卡。ui
2.去 NVIDIA官網 查看其中是否有你的顯卡型號,若是有,則說明你的顯卡是支持被 PyTorch 調用的。spa
Step2:檢查顯卡驅動版本並更新。操作系統
1.打開命令行,輸入 nvidia-smi
,查看本身的 「Driver Version」.net
注:如出現以下報錯:命令行
'nvidia-smi' 不是內部或外部命令,也不是可運行的程序
請參考文末「可能遇到的問題」進行處理。3d
通常而言,不一樣版本的CUDA要求不一樣的NVIDIA驅動版本,同時顯卡驅動版本要不低於CUDA的安裝版本。code
CUDA對顯卡驅動版本要求如圖:
好比我安裝的是 PyTorch 1.5 + CUDA 9.2 版本,要求電腦的顯卡驅動大於398.26。
2.若是驅動版本過低,在官網,下載對應的最新驅動並安裝更新。
選擇相應的顯卡型號,操做系統,下載類型和語言默認。其中的 Notebooks 是指筆記本。
以後,點擊搜索,下載最新驅動後,按照指引進行安裝便可。
Step3:驗證驅動版本及GPU是否可調用。
1.再次在終端窗口輸入 nvidia-smi
,查看最新的版本是否安裝成功。
2.進入python環境:
conda activate pytorch python
3.在python環境中:
import torch torch.cuda.is_available()
查看返回結果是不是True
可能遇到的問題:
1.執行'nvidia-smi' 命令時提示'nvidia-smi'不是內部或外部命令,也不是可運行的程序
緣由:由於它找不到該命令。這個文件是一個exe文件,通常都在下面這個文件夾中。
C:\Program Files\NVIDIA Corporation\NVSMI
因此想要使用該命令必需要能獲得這個文件夾,才能執行。
解決:添加環境變量
個人電腦->右鍵->屬性->高級系統設置->高級->環境變量->系統變量->path->編輯->新建->拷貝nvidia-smi.exe文件所在路徑(通常爲C:\\Program Files\\NVIDIA Corporation\\NVSMI
)
參考:
https://www.cnblogs.com/zhouz...
https://blog.csdn.net/shuiyix...
https://docs.nvidia.com/cuda/...