Ubuntu系统+Conda环境中降低Pytorch版本的方法
步骤1:终端激活安装的pytorch环境
conda activate pytorch
步骤2:执行以下命令
conda install pytorch=1.10.0 torchvision torchaudio cudatoolkit=11.3 -c pytorch
步骤3:检查版本信息
import torch
print(torch.__version__)
注意:pytorch=1.10.0,cudatoolkit=11.3,此处选择你需要的版本