windows10安装cuda+cndnn+pytorch

windows安装cuda+cndnn+pytorch

1. 安装cuda, 确认自己的GPU最高支持的cuda版本下载链接：https://developer.nvidia.com/cuda-toolkit-archive

help中的系统信息中可以看到

 

 

 下载对应的cuda版本（这里一定要注意，和pytorch支持的最高版本的cuda对应，比如我这里就不能选11.1，只能是11.0，因为pytorch windows版本现在只支持到11.0）

直接官网下载，然后无脑安装就可以，这里下载的是在线安装的版本

2. 安装cndnn

直接官网下载：https://developer.nvidia.com/rdp/cudnn-archive 下载对应版本的cndnn

 

 

 

 完成！

3. 安装pytorch

下载链接：https://download.pytorch.org/whl/torch_stable.html