注意:30系列显卡只支持CUDA11.0及以上版本加速,因此本文CUDA选择v11.5
软件包搭配:win11 + RTX3060 + CUDA=11.5 + CuDnn=8.3.2 + python=3.10.14 + pytorch=1.12.1
0. CUDA及CuDNN安装省略
1. 先创建一个环境
conda create -n Pytorch_0 python=3.6
创建成功,如下:
2. 安装对应版本的Pytorch,代码:
conda install pytorch==1.12.1 torchvision==0.13.1 torchaudio==0.12.1 cudatoolkit=11.3 -c pytorch
结果显示,Python版本不合适(本版本的Pytorch,需要python=3.10-3.11版本之间)
重新安装其他版本的python