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
第二种:利用win10左下角的搜索功能NVIDIA Control Panel打开后——帮助——系统信息——组件:
然后可以看到我的笔记本支持的是11.2版本的cuda,cuda版本是向下兼容的,这里查询到的是电脑支持的最高的cuda版本。
二、cuda的下载安装
1.官网下载
链接: https://developer.nvidia.com/cuda-toolkit-archive
按照下面的步骤依次选择:
下载之后的文件,双击它进行安装:
2. CUDA安装
因为我已经在电脑上安装过啦,所以有些截图不是自己的,就在网上找的,除了版本与我自己的不一致其它的没有影响。
2.1 第一个界面:这个路径默认就好了,只是个临时解压目录,安装结束后,会自动删除的,不用管的,点击ok。
2.2 第二个界面:要经历小小的等待,
第三个界面:选择自定义安装界面
第四个界面:选择安装位置,这里不要做任何的修改,默认就好啦,然后下一步。
等待一段时间,点击关闭就好啦!
然后我们去终端测试一下,使用nvcc -V或者nvcc --version两个命令都可以:
三、环境变量的配置
先查看一下,cuda安装完成后下面会默认有的:
进入到环境变量的编辑:
四、cuDNN的下载安装
1. 官网下载
链接: https://developer.nvidia.com/rdp/cudnn-archive
需要注意的是cuDNN下载可能需要登录NVIDIA,没有的话需要创建账号哦!
2. cuDNN安装
下过来就是这样一个文件夹:
打开是这样的,有三个文件夹:
然后呢将上面cuDNN中的三个文件夹分别copy到cuda文件夹中就行啦。
到这里cuda以及cuDNN的安装配置就大功告成啦!
五、pytorch的下载安装
1. 官网下载
链接: https://pytorch.org/get-started/locally/
注意这里的箭头所指的一些选项,然后复制使用Run this Command所给的指令,在dos中直接运行。
2.pytorch安装
例如下面这样:
另外呢,这里介绍一下pytorch的镜像安装:
下载链接:https://download.pytorch.org/whl/torch_stable.html注意:下面这些参数一定要跟自己电脑上已有的信息对应哦:
下过来文件长这个样子:
然后在dos界面安装一下:
镜像安装缺点就是torch依赖的库需要单独安装,例如torchvision,torchaudio等。完成之后测试一下:
总结
上面就是cuda+cudnn+pytorch安装的全部过程啦,我自己配置这些环境已经很熟悉了,但是偶尔还是会遇到错误,所以遇到错误查一查就ok啦,网上也有很多资源,这里就分享一下自己的配置过程供大家参考。