Ubuntu上如何安装PyTorch的官方教程

在Ubuntu上安装PyTorch的官方教程如下：

安装前准备

1. 更新系统包列表：

sudo apt update

2. 安装Python和pip（如果尚未安装）：

sudo apt install python3 python3-pip

3. 安装虚拟环境（可选，但推荐）：

sudo apt install python3-venv

安装PyTorch

1. 创建一个新的虚拟环境（可选）：

python3 -m venv pytorch-env

2. 激活虚拟环境（如果创建了虚拟环境）：

source pytorch-env/bin/activate

3. 根据您的CUDA版本选择合适的PyTorch安装命令。您可以在PyTorch官方网站找到适用于Ubuntu的安装命令：PyTorch官方安装命令

• 无GPU支持（CPU版本）：

pip install torch torchvision torchaudio

• 有GPU支持（例如CUDA 11.1）：

pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu111/torch_stable.html

验证安装

在Python中输入以下代码来验证PyTorch是否成功安装：

import torch
print(torch.__version__)

如果安装成功，您将看到PyTorch的版本号。

请注意，以上信息仅供参考，具体安装步骤可能会因Ubuntu版本、Python版本和CUDA版本的不同而有所差异。建议您在安装前仔细阅读PyTorch官方文档，并根据自己的实际情况进行调整。