win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程

win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程

    • 安装CUDA
    • 下载cudnn
    • 下载并安装pytorch
    • 思考

以下记录本人坎坷的安装过程,以及一些解决方法。希望大家都有美好心情呀!

安装CUDA

  • 检查自己电脑的NVIDIA ,桌面右键->NVIDIA控制面板->帮助->系统信息->组件
    在这里插入图片描述

2.在官网(https://developer.nvidia.com/cuda-downloads)选择适合自己电脑的cuda版本,比自己的cuda驱动适配版本低就可以。
特别注意: 这个链接默认进入的是最新版本的下载界面,需要自己找适合的版本的!!!

  • 如果不慎安装版本错误(安装正确可跳过): 先卸载错误的再安装正确的 。找到控制面板中程序卸载页面,只保留如下框选部分,其他NVIDIA相关全部卸载:
    win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第1张图片
    再把NVIDIA GPU Computing Toolkit(C:\Program Files\NVIDIA GPU Computing Toolkit)文件夹删除,另一个别删!(有的教程会让删但后果是什么不知道,没尝试)。有闲情逸致的话,可以再清理一下注册表。然后正常安装新的就行了。

2.然后安装,提醒最好装在默认路径下,因为我发现即使修改了安装路径,不一会也会自动转移到默认路径。选择自定义选项
win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第2张图片
别选这个
win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第3张图片

  1. 傻瓜安装即可。如果不成功,就卸载重装,终会成功。
  2. 添加系统环境变量,理论上有这四个:
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\lib\x64
    缺啥补啥,前两个一般会给你配好。
    5.测试是否成功。在cmd输入nvcc -V,看是否出现cuda版本信息。

下载cudnn

链接:https://developer.nvidia.com/rdp/cudnn-download

1.需要注册并填写问卷,耐心就好了。下载好之后是个压缩包,解压后,将里面对应的bin,lib,include内的文件(注意是文件不是目录)复制粘贴(不是替换)到CUDA安装目录的对应相关文件夹下(本人是默认路径:C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2)。
2.测试是否成功。在C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\demo_suite下打开powershell,输入 .\bandwidthTest.exe
成功会如下图所示:
win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第4张图片
到这,只是成功了一小步…后面才是大头

下载并安装pytorch

注意: 这里需要提前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\用户名)可以三个通道,两个清华的一个默认的。
win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第5张图片
再添加清华的pytorch镜像。

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/

2.进入pytorch官网 https://pytorch.org/,选择自己合适的信息(CUDA版本和之前下载的版本要一样),获取命令行,记住去掉-c pytorch(不然就是默认从官网下载会失败)
win10(python3.9)安装cuda10.2+cudnn+pytorch超详细过程_第6张图片
将命令行复制到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相关内容

要保持愉快心情哦!

你可能感兴趣的:(python,机器学习,pytorch,深度学习,python)