3050显卡配置pytorch-gpu(cuda11.3版本)

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

3050显卡配置pytorch-gpu(cuda11.3版本)_第1张图片

3、用print(torch.cuda.is_available())检查是否完成

你可能感兴趣的:(pytorch,python,深度学习)