PyTorch 安装¶

PyTorch 是一个专为机器学习设计的 Python 编程语言框架。它包含一套用于处理模型的工具，常用于自然语言处理、计算机视觉及其他类似领域。您可以按照本指南在服务器上手动安装它。

Linux 上的 PyTorch 安装¶

本教程适用于以下操作系统：Ubuntu 22.04，并已在以下 Python 版本中验证：Python 3.10。

备注

如果您计划使用 GPU 加速，请根据此教程安装 NVIDIA 驱动程序和 CUDA。

安装 Python：
```
sudo apt install python3.10
```
在 Ubuntu 22.04 中，此版本默认已安装，因此我们不建议安装更新的版本。
创建 Python 虚拟环境：
```
python3 -m venv venv
```
激活虚拟环境：
```
source venv/bin/activate
```
成功激活后，提示符中将包含括号内的虚拟环境名称：
```
(venv) user@49069:~$
```
备注

您可以创建任意数量的虚拟环境，并安装不同的库（包括同时安装，但有时这可能会导致冲突）。

在虚拟环境中安装 PyTorch 库：

pip3 install torch torchvision torchaudio

验证 PyTorch 安装：

为此，请运行 Python 控制台命令 python，然后运行以下程序：

import torch
x = torch.rand(5, 3)
print(x)

如果安装成功，您将收到以下输出：

(venv) user@49069:~$ python
Python 3.10.12 (main, Nov 20 2023, 15:14:05) [GCC 11.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
 >>> import torch
 >>> x = torch.rand(5, 3)
 >>> print(x)
tensor([[0.80, 0.04, 0.6],
     [0.32, 0.59, 0.7],
     [0.8, 0.70, 0.25],
     [0.40, 0.9, 0.9],
     [0.8, 0.15, 0.5]])

验证 PyTorch 库是否使用 CUDA：

为此，请在 Python 控制台中运行以下程序：
```
import torch
torch.cuda.is_available()
```
如果 PyTorch 可以在 GPU 模式下工作，输出将为：
```
 >>> import torch
 >>> torch.cuda.is_available()
True
```

要退出虚拟环境，请使用以下命令：

deactivate