【這其實是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
![](https://img.laitimes.com/img/__Qf2AjLwojIjJCLyojI0JCLiAzNfRHLGZkRGZkRfJ3bs92YsYTMfVmepNHL9UlMaVnRtNGaOhlYo50MMBjVtJWd0ckW65UbM5WOHJWa5kHT20ESjBjUIF2X0hXZ0xCMx81dvRWYoNHLrdEZwZ1Rh5WNXp1bwNjW1ZUba9VZwlHdssmch1mclRXY39CXldWYtlWPzNXZj9mcw1ycz9WL49zZuBnL0UDNxAjMxYTMzAjMwEjMwIzLc52YucWbp5GZzNmLn9Gbi1yZtl2Lc9CX6MHc0RHaiojIsJye.png)
對比之下,我的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:
(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()就行