Windows端CUDA11.3+CUDNN+pytorch环境搭建

1、显卡驱动的安装

最近在学习pytorch深度学习遇到很多的坑环境配置也出现过问题忍不住和大家进行分享现在把环境搭建过程分享给大家。

1.1、查看自己的显卡。

具体操作我的电脑-——属性——设备管理器——显示适配器

1.2、驱动的下载、安装。

在找到自己的显卡后去NVIDIA官网下载安装驱动

 如果成功显示显卡数据恭喜。右上角的11.6说明我的显卡驱动最高支持11.6的cuda。但我肯定不能安装11.6因为pytorch最高才支持11.3。

2、CUDA11.3下载安装

CUDA官网

在这里先等一下博主当时下载好10.2的CUDA然后打开torch的官网后发现;

 CUDA10.2现在竟然不能够在Windows上安装了我只好又重新下了一个CUDA11.3。

下载完后开始安装这是我安装10.2的过程这个无所谓默认的话一路nest就行。

 4、pytorch安装

 创建一个新的anaconda环境以防万一把自己的base环境搞崩了cd到你的环境中然后输入图中conda install......这一行。

检测安装

# 依次输入
import torch
torch.cuda.is_available()

 显示True表明下载完成可以开始你的深度学习之旅了