我的笔记本是NVIDIA GeForce RTX 3050 Laptop GPU的,原先安装cuda10.2不能使,查了3050芯片安装哪个版本:
CUDA Toolkit 11.1.1 cuDNN v8.1.1 (Feburary 26th, 2021), for CUDA 11.0,11.1 and 11.2
pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0
参考Windows 11+RTX3050Ti安装CUDA、cuDNN与Pytorch的正确版本-CSDN博客
接下来是具体操作:
先下载安装包:
1.选择CUDA Toolkit 11.1.1
https://developer.nvidia.com/cuda-toolkit-archive
Win-x86-10-exe
2.选择cuDNN v8.1.1 ...... for CUDA 11.0,11.1 and 11.2 ---cuDNN ......Windows(x86)
https://link.zhihu.co