ICode9

精准搜索请尝试: 精确搜索
首页 > 其他分享> 文章详细

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

2021-10-23 15:02:42  阅读:1630  来源: 互联网

标签:available false torch pytorch CUDA 版本 安装 下载


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

(N卡驱动更新)

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

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

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

二)安装流程 

        1,安装anaconda

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

        有需要可以查看CSDNicon-default.png?t=L9C2https://mp.csdn.net/mp_blog/creation/editor/120920748

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

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

       

        2,环境管理:

        在上图(base)后面输入conda create -n pytorch python=3.7,然后如下图,输入y开始下载这些包;

下载完成后,输入

activate pytorch

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

        3,安装CUDA

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

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

 

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

(我第一次直接装了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。

 

        4,安装pytorch

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

 

        复制上图中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,下载可能会比较慢,但是不能直接使用代理下载,会被阻止(报一堆红色错误),无法下载,建议选个人少的时间段

  

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

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

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

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

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

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

然后输入

python
import torch
torch.cuda.is_available()

终于,返回了true,如下图

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

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

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

标签:available,false,torch,pytorch,CUDA,版本,安装,下载
来源: https://blog.csdn.net/qq_42599206/article/details/120920195

本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享;
2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关;
3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关;
4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除;
5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

专注分享技术,共同学习,共同进步。侵权联系[81616952@qq.com]

Copyright (C)ICode9.com, All Rights Reserved.

ICode9版权所有