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

 

原文地址:https://www.cnblogs.com/shanyr/p/14088631.html