安裝gpu版pytorch後torch.cuda.is_available() 是 False

說明:torch.cuda.is_available()這個指令的做用是看你電腦的 GPU 可否被 PyTorch 調用。
若是返回的結果是 False,能夠按照如下過程進行排查。html

Step1:確認硬件支持,確認你的 GPU是否支持 CUDA(是否支持被 PyTorch 調用)python

1.肯定計算機中是不是獨立顯卡,是不是 NVIDIA 顯卡。能夠從 任務管理器 或者 設備管理器 來查看顯卡的型號。性能

  • 任務管理器->性能->GPU:

image.png

  • 個人電腦->屬性->設備管理器->顯示適配器:

image.png

有「NVIDIA GeForce 840M」字樣信息表示有獨立顯卡。ui

2.去 NVIDIA官網 查看其中是否有你的顯卡型號,若是有,則說明你的顯卡是支持被 PyTorch 調用的。
image.pngspa

Step2:檢查顯卡驅動版本並更新。操作系統

1.打開命令行,輸入 nvidia-smi,查看本身的 「Driver Version」
image.png.net

注:如出現以下報錯:命令行

'nvidia-smi' 不是內部或外部命令,也不是可運行的程序

請參考文末「可能遇到的問題」進行處理。3d

通常而言,不一樣版本的CUDA要求不一樣的NVIDIA驅動版本,同時顯卡驅動版本要不低於CUDA的安裝版本。code

CUDA對顯卡驅動版本要求如圖:
image.png

好比我安裝的是 PyTorch 1.5 + CUDA 9.2 版本,要求電腦的顯卡驅動大於398.26。

2.若是驅動版本過低,在官網,下載對應的最新驅動並安裝更新。

選擇相應的顯卡型號,操做系統,下載類型和語言默認。其中的 Notebooks 是指筆記本。
image.png

以後,點擊搜索,下載最新驅動後,按照指引進行安裝便可。

Step3:驗證驅動版本及GPU是否可調用。

1.再次在終端窗口輸入 nvidia-smi,查看最新的版本是否安裝成功。
2.進入python環境:

conda activate pytorch
python

3.在python環境中:

import torch
torch.cuda.is_available()

查看返回結果是不是True
image.png

可能遇到的問題:
1.執行'nvidia-smi' 命令時提示'nvidia-smi'不是內部或外部命令,也不是可運行的程序

緣由:由於它找不到該命令。這個文件是一個exe文件,通常都在下面這個文件夾中。

C:\Program Files\NVIDIA Corporation\NVSMI

因此想要使用該命令必需要能獲得這個文件夾,才能執行。

解決:添加環境變量

個人電腦->右鍵->屬性->高級系統設置->高級->環境變量->系統變量->path->編輯->新建->拷貝nvidia-smi.exe文件所在路徑(通常爲C:\\Program Files\\NVIDIA Corporation\\NVSMI
image.png

參考:
https://www.cnblogs.com/zhouz...
https://blog.csdn.net/shuiyix...
https://docs.nvidia.com/cuda/...

相關文章
相關標籤/搜索