第二篇:win10下CUDA+cuDNN+pytorch的安装配置

win10下cuda+cudnn+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啦,网上也有很多资源,这里就分享一下自己的配置过程供大家参考。

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值