win10下cuda+cudnn+pytorch的安装配置

其实网上已经有很多关于这方面的博文,自己也已经安装过无数次了,所以这里也来详细记录一下完整环境配置过程,希望可以帮助到刚接触到这方面的新手小白,每一步我都会附上截图的,所以不要怕不会操作哦。



文章目录

  • win10下cuda+cudnn+pytorch的安装配置
  • 一、首先查看电脑可适配的cuda版本
  • 二、cuda的下载安装
  • 1.官网下载
  • 2. CUDA安装
  • 三、环境变量的配置
  • 四、cuDNN的下载安装
  • 1. 官网下载
  • 2. cuDNN安装
  • 五、pytorch的下载安装
  • 1. 官网下载
  • 2.pytorch安装
  • 总结



一、首先查看电脑可适配的cuda版本

第一种:win+R——输入cmd打开dos界面,输入nvidia-smi

cuda测试程序 pytorch cuda cudnn pytorch_cuda测试程序 pytorch


第二种:利用win10左下角的搜索功能NVIDIA Control Panel打开后——帮助——系统信息——组件:

cuda测试程序 pytorch cuda cudnn pytorch_官网_02


然后可以看到我的笔记本支持的是11.2版本的cuda,cuda版本是向下兼容的,这里查询到的是电脑支持的最高的cuda版本。


二、cuda的下载安装

1.官网下载

链接: https://developer.nvidia.com/cuda-toolkit-archive

cuda测试程序 pytorch cuda cudnn pytorch_下载安装_03


按照下面的步骤依次选择:

cuda测试程序 pytorch cuda cudnn pytorch_深度学习_04


下载之后的文件,双击它进行安装:

cuda测试程序 pytorch cuda cudnn pytorch_cuda测试程序 pytorch_05

2. CUDA安装

因为我已经在电脑上安装过啦,所以有些截图不是自己的,就在网上找的,除了版本与我自己的不一致其它的没有影响。

2.1 第一个界面:这个路径默认就好了,只是个临时解压目录,安装结束后,会自动删除的,不用管的,点击ok。

cuda测试程序 pytorch cuda cudnn pytorch_下载安装_06


2.2 第二个界面:要经历小小的等待,

cuda测试程序 pytorch cuda cudnn pytorch_下载安装_07


cuda测试程序 pytorch cuda cudnn pytorch_官网_08


第三个界面:选择自定义安装界面

cuda测试程序 pytorch cuda cudnn pytorch_cuda测试程序 pytorch_09


cuda测试程序 pytorch cuda cudnn pytorch_深度学习_10


第四个界面:选择安装位置,这里不要做任何的修改,默认就好啦,然后下一步。

cuda测试程序 pytorch cuda cudnn pytorch_深度学习_11


等待一段时间,点击关闭就好啦!

cuda测试程序 pytorch cuda cudnn pytorch_官网_12


然后我们去终端测试一下,使用nvcc -V或者nvcc --version两个命令都可以:

cuda测试程序 pytorch cuda cudnn pytorch_官网_13



三、环境变量的配置

先查看一下,cuda安装完成后下面会默认有的:

cuda测试程序 pytorch cuda cudnn pytorch_安装配置_14


进入到环境变量的编辑:

cuda测试程序 pytorch cuda cudnn pytorch_下载安装_15



四、cuDNN的下载安装

1. 官网下载

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

需要注意的是cuDNN下载可能需要登录NVIDIA,没有的话需要创建账号哦!

cuda测试程序 pytorch cuda cudnn pytorch_官网_16

2. cuDNN安装

下过来就是这样一个文件夹:

cuda测试程序 pytorch cuda cudnn pytorch_安装配置_17


打开是这样的,有三个文件夹:

cuda测试程序 pytorch cuda cudnn pytorch_官网_18


然后呢将上面cuDNN中的三个文件夹分别copy到cuda文件夹中就行啦。

cuda测试程序 pytorch cuda cudnn pytorch_官网_19


到这里cuda以及cuDNN的安装配置就大功告成啦!


五、pytorch的下载安装

1. 官网下载

链接: https://pytorch.org/get-started/locally/

注意这里的箭头所指的一些选项,然后复制使用Run this Command所给的指令,在dos中直接运行。

cuda测试程序 pytorch cuda cudnn pytorch_官网_20

2.pytorch安装

例如下面这样:

cuda测试程序 pytorch cuda cudnn pytorch_深度学习_21


另外呢,这里介绍一下pytorch的镜像安装:

下载链接:https://download.pytorch.org/whl/torch_stable.html注意:下面这些参数一定要跟自己电脑上已有的信息对应哦:

cuda测试程序 pytorch cuda cudnn pytorch_下载安装_22


下过来文件长这个样子:

cuda测试程序 pytorch cuda cudnn pytorch_安装配置_23


然后在dos界面安装一下:

cuda测试程序 pytorch cuda cudnn pytorch_官网_24


镜像安装缺点就是torch依赖的库需要单独安装,例如torchvision,torchaudio等。完成之后测试一下:

cuda测试程序 pytorch cuda cudnn pytorch_cuda测试程序 pytorch_25

总结

上面就是cuda+cudnn+pytorch安装的全部过程啦,我自己配置这些环境已经很熟悉了,但是偶尔还是会遇到错误,所以遇到错误查一查就ok啦,网上也有很多资源,这里就分享一下自己的配置过程供大家参考。