天天看點

關于torch.cuda.is_available()傳回false的問題解決問題:nvidia顯示卡驅動和cuda版本能對應,但是torch.cuda.is_available()還是傳回false。解決:核查cuda版本和pytorch版本是否能夠比對!!!

【這其實是Windows下安裝Pytorch1.6.0、cuda10.2和cudnn踩坑記錄!】

問題:nvidia顯示卡驅動和cuda版本能對應,但是torch.cuda.is_available()還是傳回false。

解決:核查cuda版本和pytorch版本是否能夠比對!!!

【注意!!!】

1、Pytorch版本要和cuda版本相對應!!!

2、Nvidia驅動的版本也要和cuda版本相一緻!

說說我的踩坑記錄:我的Nvidia驅動版本是398.75,裝,的cuda是9.2版本的,按理來說cuda和Nvidia驅動版本是比對上的。但是!我的torch.cuda.is_available()傳回的是False,是以就找了一圈文章下來。找不到具體解決辦法,然後我才考慮到是不是還有什麼版本需要比對啊,我又考慮過是不是Python3.8版本太高了,而後才考慮到了Pytorch的版本問題。

于是找到了文章https://blog.csdn.net/weixin_43839245/article/details/108274605

關于torch.cuda.is_available()傳回false的問題解決問題:nvidia顯示卡驅動和cuda版本能對應,但是torch.cuda.is_available()還是傳回false。解決:核查cuda版本和pytorch版本是否能夠比對!!!

對比之下,我的pytorch是1.6.0版本,隻支援10.1和10.2,是以安裝的cuda 9.2版本不可用。

是以現在的解決方法就是

1. 給pytorch降低版本

2. 安裝cuda10.2并且更新Nvidia驅動

首先,我嘗試了把pytorch降到1.4.0,結果還是傳回false,也不知道是不是哪個步驟錯了。

然後就嘗試方法二:安裝cuda 10.2

(1) 更新Nvidia驅動

我自己直接在Nvidia官網下載下傳了最新版本的驅動(460.89),驗證驅動安裝成功是否:先cmd,再nvidia-smi,顯示了驅動資訊就表示驅動安裝成功了。

驅動和cuda對應版本的如下https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html:

關于torch.cuda.is_available()傳回false的問題解決問題:nvidia顯示卡驅動和cuda版本能對應,但是torch.cuda.is_available()還是傳回false。解決:核查cuda版本和pytorch版本是否能夠比對!!!

(2)安裝cuda10.2

  直接在官方下載下傳就好了,我下載下傳的二點多個G,下載下傳出了點問題,直接是以2kb下載下傳下來了,反複試了很多次還是這樣。然後我就打開了VPN,下載下傳就順利了。

cuda安裝成功的驗證:直接cmd下輸入nvcc -V,能顯示你的cuda版本資訊就是安裝成功啦。再然後安裝cudnn就是把幾個檔案夾裡的内容挪一下,比較簡單,這裡就不講了。

(3)驗證torch.cuda.is_available()

先python,或者打開你裝了pytorch的環境,我的pytorch直接安裝在了base環境裡。

然後import torch 【torch.__version__ 可以直接查詢安裝的pytorch版本】

最後torch.cuda.is_available()就行

關于torch.cuda.is_available()傳回false的問題解決問題:nvidia顯示卡驅動和cuda版本能對應,但是torch.cuda.is_available()還是傳回false。解決:核查cuda版本和pytorch版本是否能夠比對!!!

繼續閱讀