- 安装CUDA
- 下载cudnn
- 下载并安装pytorch
- 思考
以下记录本人坎坷的安装过程,以及一些解决方法。希望大家都有美好心情呀! 安装CUDA
- 检查自己电脑的NVIDIA ,桌面右键->NVIDIA控制面板->帮助->系统信息->组件
2.在官网(https://developer.nvidia.com/cuda-downloads)选择适合自己电脑的cuda版本,比自己的cuda驱动适配版本低就可以。
特别注意: 这个链接默认进入的是最新版本的下载界面,需要自己找适合的版本的!!!
- 如果不慎安装版本错误(安装正确可跳过): 先卸载错误的再安装正确的 。找到控制面板中程序卸载页面,只保留如下框选部分,其他NVIDIA相关全部卸载:
再把NVIDIA GPU Computing Toolkit(C:Program FilesNVIDIA GPU Computing Toolkit)文件夹删除,另一个别删!(有的教程会让删但后果是什么不知道,没尝试)。有闲情逸致的话,可以再清理一下注册表。然后正常安装新的就行了。
2.然后安装,提醒最好装在默认路径下,因为我发现即使修改了安装路径,不一会也会自动转移到默认路径。选择自定义选项
别选这个
- 傻瓜安装即可。如果不成功,就卸载重装,终会成功。
- 添加系统环境变量,理论上有这四个:
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2bin
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libnvvp
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2
C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2libx64
缺啥补啥,前两个一般会给你配好。
5.测试是否成功。在cmd输入nvcc -V,看是否出现cuda版本信息。
链接:https://developer.nvidia.com/rdp/cudnn-download
1.需要注册并填写问卷,耐心就好了。下载好之后是个压缩包,解压后,将里面对应的bin,lib,include内的文件(注意是文件不是目录)复制粘贴(不是替换)到CUDA安装目录的对应相关文件夹下(本人是默认路径:C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2)。
2.测试是否成功。在C:Program FilesNVIDIA GPU Computing ToolkitCUDAv10.2extrasdemo_suite下打开powershell,输入 .bandwidthTest.exe
成功会如下图所示:
到这,只是成功了一小步…后面才是大头
注意: 这里需要提前anaconda,如果没有安装先去安装!当然你也可以用pip安装,但是不管是直接下载或者换源下载,我都失败了。conda的话,直接下载也失败,需要换源!
1.换源:此处使用该po主的方法,只是安装pytorch处借鉴了(https://blog.csdn.net/zhangxiaoduoduo/article/details/115772066)
在anaconda prompt分别输入以下命令行:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/ conda config --set show_channel_urls yes
打开.condarc文件(路径:C:Users用户名)可以三个通道,两个清华的一个默认的。
再添加清华的pytorch镜像。
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
2.进入pytorch官网 https://pytorch.org/,选择自己合适的信息(CUDA版本和之前下载的版本要一样),获取命令行,记住去掉-c pytorch(不然就是默认从官网下载会失败)
将命令行复制到prompt上运行就可以了,会下载很多包,如果某一条失败了就再运行一遍命令行。
3.下载好的包在在Anaconda的安装目录的pkgs下,也可以在prompt里写pip list查看有没有torch这个包。
4.打开个python软件测试以下import torch是否成功即可。
之前因为安装pytorch失败,一直在找原因,怀疑是python和pytorch版本不匹配,但好像现在pytorch下载不需要选择python版本了,目前两者具体如何匹配问题不清楚,但是不是导致安装失败的原因。
根据下载几次的经验来看,如果出错了,首先检查CUDA和cudnn的问题(自己电脑和cuda的版本是否匹配,cuda和cudnn的版本是否匹配。)
最后附上可能有帮助的链接,希望大家都可以顺利安装:
win10安装CUDA和cuDNN的正确姿势
WIN10环境下安装Python+CUDA+cuDNN+Anaconda+Pytorch
理清GPU、CUDA、CUDA Toolkit、cuDNN关系以及下载安装
pytorch中关于cuda相关内容
要保持愉快心情哦!



