Windows10安装CUDA和cuDNN

前言

我们都知道,CUDA和cuDNN是用来加速深度学习计算的,那么怎么在Windows上面安装CUDA以及cuDNN呢?

准备阶段

1 自己电脑上应该有Nvidia显卡,显存最好8G或者8G以上,当然4G或者6G也行,只不过就是Batch size 设置小一点
2 查看Nvidia显卡版本,一般Windows上有显卡的话,显卡驱动都是自动安装的,所以我们可以通过命令行,然后输入Nvidia-smi来查看自己电脑上的驱动版本:
在这里插入图片描述
这里我的显卡驱动版本是432.0 ,CUDA版本为10.1,这里 需要注意下,如果您的GPU在这里已经提示了CUDA Version版本,您需要在系统盘下的C:\Program Files 和 C:\Program Files (x86) 找找有没有CUDA相关的文件夹存在,如果有的话,就不用重新安装了,如果没有的话,需要手动进行安装,这里我也不确定通过nvidia-smi查看前,是否显卡已经预装了CUDA,查阅了一些资料,貌似这个查出来的版本是指该显卡支持的最高的CUDA版本,原文链接
在这里插入图片描述

安装CUDA ToolKit

点击进入
在这里插入图片描述

选择合适的版本进入,按照以下方式选择

在这里插入图片描述

下载后安装

1 同意安装
在这里插入图片描述
2 选择自定义在这里插入图片描述
3 默认所有组件,下一步在这里插入图片描述
4 记住这里的安装路径,可以自行改变,我这里由于使用的固态,所以默认就在C盘了在这里插入图片描述
5 开始安装在这里插入图片描述

开始安装cuDNN

点击进入

注意这里需要你有个Nvidia的账号使用自己的Email注册就行
在这里插入图片描述
在这里插入图片描述
下载结束后,解压
在这里插入图片描述
将这三个文件夹的内容复制到之前提示你记得三个路径,其实没改过的话默认是
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
将cudnn文件夹里三个文件里的内容复制到上面文件里相同名字的文件夹下,即可。
在这里插入图片描述

CUDA环境变量

这个会自动生成
在这里插入图片描述
再在环境变量中添加以下路径:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

也有可能在系统路径中已经有了这些路径,就不用添加了

在这里插入图片描述

这就装好了CUDA和cuDNN,这就可以后续的实验了,也可以查看cuDNN版本

记事本打开
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\include\cudnn.h
可以看到版本号
在这里插入图片描述

Pytorch

点击进入
可以下载对应CUDA10.1版本的pytorch的whl文件进行安装,因为CUDA 10.1版本在Pytorch官网上是没有的,
cu后为cuda版本,torch后为torch版本,torchvision后为torchvision版本,cp后为python版本,最后面代表windows64位
在这里插入图片描述

其他的框架可以自行安装GPU版本的
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值