@TOC
第一步
檢查cuda是否安裝,CUDA環(huán)境變量是否正確設(shè)置,比如linux需要設(shè)置在PATH,window下環(huán)境變量編輯看看,是否有CUDA
第二步,核查python中torch版本
首先查看你環(huán)境里的pytorch是否是cuda版本,我這里是conda安裝的,使用conda list查看是這樣:
圈出來的torch帶cuda,那安裝的肯定時cuda版本,但是torch.cuda.is_available()還是False,這種情況就比較費解,重點來了,我們?nèi)ズ藢ο聀ython環(huán)境中用的是什么
可以很清楚的看到這里用的是cpu版本,也就是說環(huán)境中確實存在兩個版本
所以環(huán)境使用的是cpu才導(dǎo)致加載不到gpu版本,可以卸載torch,重新安裝一下
conda uninstall pytorch
然后從官網(wǎng)重新下載一下 ,記住這里有個坑,如果你配置了多個鏡像源,這個鏡像源如果沒有g(shù)pu版本,可能還是會下載cpu版本
下載下來以后一定要再次核對一下,或者下載的時候指定下載的鏡像源
第三步,核查顯卡驅(qū)動和cuda版本是否適配
如果上述沒有問題,那么檢查驅(qū)動和cuda的版本是否適配文章來源:http://www.zghlxwxcb.cn/news/detail-685135.html
NVIDIA-smi
這兩個版本是否適配需要在官網(wǎng)查看
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html#major-components文章來源地址http://www.zghlxwxcb.cn/news/detail-685135.html
到了這里,關(guān)于【安裝GPU版本pytorch,torch.cuda.is_available()仍然返回False問題】的文章就介紹完了。如果您還想了解更多內(nèi)容,請在右上角搜索TOY模板網(wǎng)以前的文章或繼續(xù)瀏覽下面的相關(guān)文章,希望大家以后多多支持TOY模板網(wǎng)!