准备
- 桌面右键选择NVIDIA控制面板,然后依次选择帮助->系统信息->组件
- 可以看到我这台笔记本支持的cuda版本是
10.2.95
,记住这个 - 然后去nvidia官网下载对应的驱动程序包
- 然后去下载对应的cudnn
,下载cudnn需要NVIDIA账号,如果没有,建议使用163邮箱注册一个(为什么呢?因为QQ邮箱现在收不到NVIDIA的邮件,这就不知道到底是腾讯的问题还是英伟达的问题,亦或者是两者都有问题,手动狗头)
- 单独提一嘴,如果下载速度很慢的话,不要奇怪,国外的网站是这样的,想提高下载速度,请挂科学上网,或者是自己找大佬们下载好放在网盘里的资源
安装cuda
- 直接双击运行下载好的exe就行
- 基本都是下一步下一步下一步,没啥好说的
- 验证cuda,命令
nvcc -V
,看到如下图所示就行了
配置cudnn
- 解压cudnn压缩包
- 直接 Ctrl + A 全选,然后 Ctrl + C 复制,然后进入
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2
目录下(请注意根据版本等选择对应的目录,不要无脑就直接用我这个目录), Ctrl + V 粘贴
安装pytorch
- 去pytorch官网,根据操作系统,python包管理等选择对应的选项
- 复制上面选择后出现的安装命令,我这里选的是pip,命令行执行
pip install torch===1.5.1 torchvision===0.6.1 -f https://download.pytorch.org/whl/torch_stable.html
- 等待安装完成,可能会比较慢,请挂科学上网
验证pytorch
进入python交互式解释器,cmd中执行
python
命令即可参考官方示例进行验证
官方示例:1
2
3
4from __future__ import print_function
import torch
x = torch.rand(5, 3)
print(x)输出:
1
2
3
4
5tensor([[0.3380, 0.3845, 0.3217],
[0.8337, 0.9050, 0.2650],
[0.2979, 0.7141, 0.9069],
[0.1449, 0.1132, 0.1375],
[0.4675, 0.3947, 0.1426]])验证cuda
官方示例:1
2import torch
torch.cuda.is_available()输出:
1
True
验证通过,enjoy it!