1.cmd打开命令行
conda create --name py38 python=3.8
2.激活环境
conda avtivate py38
3.安装CUDA
右击 NVIDIA设置–>NVIDIA控制面板–>帮助–>系统信息–>组件
4.安装torch
pytorch官网查看版本并安装
conda install pytorch==1.12.0 torchvision==0.13.0 torchaudio==0.12.0 cudatoolkit=11.6 -c pytorch -c conda-forge
上面为torchvision最新版本,里面很多新的模型,超好用。
5.手动逐个安装
版本对应参考
whl下载连接