在Ubuntu系统中回滚PyTorch版本可以通过以下步骤进行:
如果您已经安装了PyTorch,首先需要卸载当前版本。可以使用以下命令:
pip uninstall torch torchvision torchaudio
访问PyTorch官方网站,在“Previous PyTorch Versions”部分找到与您的系统兼容的旧版本。
确定需要的版本后,使用pip或conda来安装目标版本。例如,安装PyTorch 1.9.0(与CUDA 11.5兼容):
pip install torch==1.9.0 torchvision==0.8.1 torchaudio==0.7.0
或者使用conda:
conda install pytorch==1.9.0 torchvision==0.8.1 torchaudio==0.7.0 -c pytorch
请注意,您需要将1.9.0替换为您找到的正确的版本号,而+cu115表示这个包是为CUDA 11.5编译的。在运行这些命令之前,请确保您的环境(例如conda环境)是激活的。
安装完成后,在Python中运行以下代码来验证PyTorch和CUDA是否正确安装:
import torch
print(torch.__version__)
这应该会输出PyTorch的版本,PyTorch使用的CUDA版本,以及一个布尔值,指示CUDA是否可用。
如果您使用的是conda环境,可以很容易地实现版本降级。只需指定版本即可:
conda install pytorch=1.4.1 -c soumith
以上就是在Ubuntu系统中回滚PyTorch版本的操作指南。在执行版本回滚之前,请确保备份系统和数据,以防止数据丢失。