(1)NVIDIA显卡驱动程序下载
(2)CUDA下载
1.CUDA == 10.2
CUDA Toolkit 10.2 Download | NVIDIA Developer
2.验证CUDA安装是否成功
首先重启电脑
然后在命令行输入`nvcc -V` ,显示如下为成功
(3)CuDNN下载
1. 首先要在官网注册一个账号,如果出现“验证程序加载失败”问题,解决方法如下:
英伟达账户一直登陆不进去解决办法_du2005023029的博客-CSDN博客_nvidia验证程序加载失败请检查浏览器
2. CuDNN for CUDN10.2
https://developer.nvidia.com/rdp/cudnn-download
3. 配置路径
我的电脑——>属性——>高级系统设置——>环境变量
在系统变量的Path中,双击打开添加两个路径:
一是cudnn的bin路径`C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\cudnn\bin`
二是`C:\Program Files\NVIDIA GPU ComputingToolkit\CUDA\v10.0\extras\CUPTI\libx64`
(4)Pytorch下载
Previous PyTorch Versions | PyTorch
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=10.2 -c pytorch
(5)验证
import torch
print(torch.cuda.is_available())
返回True代表配置环境成功
注:本人安装完numpy版本有问题,重新安装一下numpy就好了