安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

PS:记录一下安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

一)GPU版安装前:

1,检查电脑是否拥有nvidia显卡,没有N卡可以用CPU版本;本机以GTX1650为例。

2,检查nvidia显卡驱动版本号,打开cmd,输入命令

nvidia-smi

确保NVIDIA-SMI与Driver Version相同(一般应该都会相同emmmm),查看CUDA Version,我的是11.3

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题(N卡驱动更新)

        官网(官方 GeForce 驱动程序 | NVIDIA) ,按照自己的显卡进行选择,我这里是gtx1650,点击开始搜索。

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

 会有很多可以更新的驱动程序,如下图:

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

可以去(Release Notes :: CUDA Toolkit Documentation) 查看自己想要的CUDA对应的驱动程序的版本,如下图:(我的是CUDA11.3,对应462.89,选择大于等于462.89的最小的一个版本,大概率就是对应的CUDA的版本)

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

二)安装流程 

        1,安装anaconda

        官网(Anaconda | The World's Most Popular Data Science Platform),我安装的是Anaconda3-5.3.1-Windows-x86_64.exe(听说比较稳定)

        有需要可以查看CSDN安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题https://mp.csdn.net/mp_blog/creation/editor/120920748

        装完后,打开win键,有一个anaconda prompt

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

 点击打开,出现如下图所示(base)即代表安装成功。

       安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

        2,环境管理:

        在上图(base)后面输入conda create -n pytorch python=3.7,然后如下图,输入y开始下载这些包;安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

下载完成后,输入

activate pytorch

出现(pytorch) C:\Users\....>代表成功。

        3,安装CUDA

        官网下载即可(CUDA Toolkit 11.5 Downloads | NVIDIA Developer

        选择对应的系统,它会出现当前最新版的CUDA,我当前最新版是11.5(2021.10)

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题 

        想用旧版(推荐,较新版更稳定)的话可以点击下图中红框所选

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

(我第一次直接装了11.5,后来装pytorch的时候发现只有10.2和11.3,所以又把11.5卸载了,装了11.3)

PS:听大神说,小版本更新11.X不会影响二者配对使用,要么都是10.X或者要么都是11.X

我为了保险起见,直接选择了CUDA11.3版,nvidia驱动直接对应的也是11.3(没有兼容),pytorch版本同11.3,属实是被搞怕了,即使这样后面还是报了false,(当时差点炸开......)

CUDA安装网上有教程,可直接搜,差别影响都不大,这里不多说。装完后cmd,输入

nvcc -V

出现下图所示,代表CUDA安装成功,版本为11.3。

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题 

        4,安装pytorch

        进入官网(PyTorch),如下图按照自己的电脑需选择:

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题 

        复制上图中pip3 install torch==1.10.0+cu.........这一段代码,是代表从官网下载3个包torch,torchvision,torchaudio。

(采坑:::切记勾选pip,不要勾选conda)

打开anaconda prompt,输入activate pytorch,进入(pytorch)

        然后粘贴pip3 install torch==1.10.0.......这一段代码,然后回车运行下载即可,下载会有点慢。

        注:1,下载可能会比较慢,但是不能直接使用代理下载,会被阻止(报一堆红色错误),无法下载,建议选个人少的时间段

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题  

(我晚上九点多下载的时候只有十几K速度,第二天早上人少速度能达到1M/s,其实也不是很慢)

            2,如果使用镜像下载,请一定仔细查看源是否拥有你所需要的版本,镜像下载建议使用10.X版本,应该没啥问题,反正暂时不想试了。

(采坑:我当时使用镜像安装,下载很快,但是还是false,后来发现下载的竟然是CPU版本,emmmm,然后老实用官网下载了)

            3,使用conda下载,仍然是报false错误,我也不知为何,反正版本啥的都匹配,(版本输出是对的,非CPU版本)就是一直报那个false,emmmm,暂时先不考虑这个了,直接上pip完事。

PS:下载的时候遇到了一个奇怪的事,它竟然没有torchaudio===0.10.0+cu113(都是一个官网,就离谱)

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

然后删掉了“+cu113",输入(仍然是”(pytorch)"环境下,然后出现这个代表下载成功。

然后输入安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

python
import torch
torch.cuda.is_available()

终于,返回了true,如下图

安装pytorch(GPU版)时遇到的关于torch.cuda.is_available()报false的问题

但是呢,更离谱的事来了,numpy出错了,(emmm,虽然暂时不影响使用,但是import torch后面出现这个很难受)

所以安装下numpy,输入代码conda install numpy,然后下载即可(如果电脑之前本身安装了numpy,可以先卸载再安装,卸载numpy代码为conda uninstall numpy

然后再次验证,就会OK了。

上一篇:CUDA 3D convolution


下一篇:cuda编程基础