Windows10+CUDA9.2+CUDNN配置pytorch(花了两天时间走了不少坑,即使现在已经晚上两点过了,但还是把这两天的总结一下)

环境说明

  • Windows10

  • Python3.6.5

  • CUDA9.2.217

  • pycharm
    最后安装的版本

  • anaconda3

  • cuda_9.2.148

  • cudnn-9.2-windows-7.2.138

步骤

因为之前已经配置好了python的环境(这里不说明安装配置python环境等):

  1. anaconda3安装(这个很简单,选择对应的Python环境就可以)
    需要注意的是,环境变量的配置问题以及添加国内镜像
    我配置了这些,参考别人的解决办法(有的只配置了两个)
    最简单的是在安装anaconda的时候默认勾选加入环境变量(建议还是使用自定义)Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    配置完成之后,可以在命令行输入conda --version来检查是否成功,或者命令行输入python也可以
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

  2. 选择电脑适合的CUDA进行安装
    首先要查看自己的电脑有没有独立显卡(在计算机管理-设备管理器-显示适配器中可以查看)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    同时一定要自己去看本机适合的CUDA版本(我之前就是版本弄错了,一直没有注意到,浪费了很多时间,同时如果不使用多个cuda,最好卸载掉下错的那个以及删除配置的环境变量)。打开NVIDIA的控制面板,在菜单栏的帮助一栏中查看系统信息-显示组件:我的就是CUDA9.2.217
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    找到了自己合适的版本之后,就可以去官网下载相应的CUDA了(点击此处)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    接下来就是一系列next的安装步骤了(安装anaconda3的位置我没有改,默认系统安装,方便以后查找)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    最后,检查是否成功:命令行输入nvcc -V,显示了版本信息则正确。
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

  3. 安装相应的CUDNN
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

    安装CUDNN必须和CUDA的版本保持一致,下载的时候很慢,也有可能下载不成功,这里的方法是点击右键复制链接,使用迅雷下载。下载成功解压之后将CUDNN目录下的lib,bin,include复制到CUDA\v9.2目录下,替换掉原来的文件。

  4. 安装pytorch
    官网下载地址
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    不知道为什么Python3.6+CUDA9.2不配拥有直接的下载链接了,使用命令行下载也只会报错:在镜像中找不到对应的包。没办法!又看见了别人使用的一个办法:离线下载.whl文件(不过需要自己知道自己需要下载的版本)
    Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)
    同样下载的时候使用复制链接到迅雷下载会相对快很多,也不会下载失败。我的版本是Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

8.离线安装.whl文件
命令行进入到存放上述两个whl文件的目录,使用pip来安装(安装之前可以先查看一下是否安装了wheel,没有的话需要安装:pip install wheel)直接命令行输入:pip install whl’s name.whl(一定要先安装torch,然后再安装torchvision).
最后,查看是否成功:点击开始-打开anaconda prompt:
Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

遇到的错误以及参考链接

  • 安装anaconda:https://www.zhihu.com/question/58033789/answer/254673663
  • 清华大学开源镜像网:https://mirrors.tuna.tsinghua.edu.cn/
  • 检查显卡支持的cuda版本:https://jingyan.baidu.com/article/6fb756ec4fabc4241858fbf7.html
  • 参考安装过程:https://blog.csdn.net/Mind_programmonkey/article/details/99688839/(这位作者写的很详细,由于后面版本不一样,也遇到了许多不同的问题)
  • Windows10+CUDA9.2+CUDNN配置pytorch(GPU)(花了两天时间走了不少坑)

可以睡觉了!!!

相关文章:

  • 2021-09-15
  • 2021-06-23
  • 2022-12-23
  • 2022-12-23
  • 2021-08-25
  • 2021-08-03
  • 2021-04-22
  • 2021-12-04
猜你喜欢
  • 2021-06-13
  • 2021-12-31
  • 2021-06-20
  • 2022-01-21
  • 2021-07-31
  • 2021-10-04
  • 2022-01-19
相关资源
相似解决方案