第一步:激活目标环境
source activate py11
第二步:安装pytorch(cpu版本)
conda list(查看pytorch)
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch(可以改为清华源)
第三步:手动安装gpu版本
清华源链接:https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/
(1)找到对应的GPU版本(主要是cuda版本)(还包括附属的torchaudio和torchvision)
(2)下载后,激活conda 环境(强调可以重复确认一下!), cd 到下载的文件目录,安装
conda install --offline pytorch-1.8.0-py3.8_cuda10.2_cudnn7.6.5_0.tar.bz2
conda install --offline torchvision-0.11.0-py38_cu102.tar.bz2
(3)conda list查看是否已经替换掉
参考链接:http://t.csdnimg.cn/HDl7K
失败记录:这个项目标注python3.11和pytorch1.8,但是好像1.8并不支持python3.11
修改:重新安装虚拟环境
conda install pytorch=1.8.0 torchvision torchaudio cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/
记得每次删除pytorch重装是,要clean
conda clean --all