Ubuntu系统+Conda环境中降低Pytorch版本的方法

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，此处选择你需要的版本