1.下载cuda和cudnn
cuda下载网址:(version:11.8)
https://developer.nvidia.com/cuda-11-8-0-download-archive
cudnn下载网址:
https://developer.nvidia.com/cudnn
二者版本一定要对应!!
cuda按照安装导航一步一步点同意,cudnn下载解压后放到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8目录下:
然后配置环境如图:
2.下载pytorch
下载网址:
https://pytorch.org/get-started/locally/
3.出现问题
我这样下载以后出现了这样的报错: