Ubuntu如何更换 PyTorch 版本

最新推荐文章于 2025-04-08 18:10:04 发布

玩人工智能的辣条哥

最新推荐文章于 2025-04-08 18:10:04 发布

阅读量1.5k

点赞数 4

分类专栏： Ubuntu Python 文章标签： ubuntu pytorch linux

版权

82 篇文章

订阅专栏

20 篇文章

订阅专栏

本文指导用户在Ubuntu22.04系统中，如何不升级CUDA版本而安装针对CUDA11.5编译的PyTorch旧版，包括卸载原有版本、查找并安装指定版本以及验证安装过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

环境：

Ubuntu22.04

WLS2

Ubuntu如何更换 PyTorch 版本考虑安装一个为 CUDA 11.5 编译的 PyTorch 版本。如何安装旧版本

决定不升级CUDA版本，而是使用一个与CUDA 11.5兼容的PyTorch版本，您可以按照以下步骤来安装一个较旧版本的PyTorch。

卸载当前的PyTorch版本：首先，您需要卸载当前安装的PyTorch，以避免任何潜在的版本冲突。在命令行中运行以下命令：
```
pip uninstall torch torchvision torchaudio
```

在这里插入图片描述

查找合适的PyTorch版本：接下来，您需要查找一个为CUDA 11.5编译的PyTorch版本。您可以访问PyTorch官方网站，并在“Previous PyTorch Versions”部分找到与CUDA 11.5兼容的版本。
安装兼容的PyTorch版本：在确定了哪个PyTorch版本与CUDA 11.5兼容之后，您可以使用pip或conda来安装。例如，如果您找到PyTorch版本1.9.0是与CUDA 11.5兼容的，您可以使用以下命令安装：

使用pip的话：
install pytorch1.12.1 torchvision0.13.1 cudatoolkit=11.3 -c pytorch

pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html

这里，-f https://download.pytorch.org/whl/torch_stable.html会告诉pip使用官方的PyTorch资源库来找到与CUDA 11.5兼容的PyTorch版本。

在这里插入图片描述

请注意，您需要将`1.9.0`替换为您找到的正确的版本号，而`+cu115`表示这个包是为CUDA 11.5编译的。在运行这些命令之前，请确保您的环境（例如conda环境）是激活的。

python -c "import torch; print(torch.__version)”

在这里插入图片描述

这应该会输出PyTorch的版本，PyTorch使用的CUDA版本，以及一个布尔值，指示CUDA是否可用。

请确保在您的工作环境或虚拟环境中执行这些步骤，避免对系统级别的Python环境产生影响。