1、显卡3050对应的CUDA最高支持版本可通过NVIDIA控制面板系统信息的组件里面查看可支持的最高版本。
2、通过以下网址,下载CUDA和CUDNN,我这里下载了11.3版本的,安装就按别人的教程来。
CUDA Toolkit Archive | NVIDIA Developer
CUDA Deep Neural Network (cuDNN) | NVIDIA Developer
3、最关键的一步!配置conda的环境。因为下载的为11.3的CUDA,所以必须根据cuda11.3的指令来下载,我这里用的命令为以下两条。
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 -c http://mirrors.aliyun.com/anaconda/cloud/pytorch/win-64/
conda install cudatoolkit=11.3 -c http://mirrors.aliyun.com/anaconda/pkgs/main
3、用print
(torch.cuda.is_available())检查是否完成