ubuntu安装gpu版本pytorch教程

### Ubuntu 上安装 GPU 版本 PyTorch 的教程

#### 准备工作
为了确保顺利安装，在开始之前确认系统已经正确配置了 NVIDIA 显卡驱动以及 CUDA 工具包。对于 Intel CPU 和 Ubuntu 20.04 系统，选择合适的 PyTorch 安装选项至关重要[^1]。

#### 使用 Anaconda 进行安装
一种推荐的方法是通过 Anaconda 来管理 Python 环境并安装 PyTorch 及其依赖项。可以执行如下命令来创建一个新的 Conda 环境，并指定所需的 CUDA 版本：

conda create -n pytorch_env python=3.8
conda activate pytorch_env

接着利用 Conda 命令直接安装带有特定 CUDA 支持的 PyTorch:

conda install pytorch torchvision torchaudio cudatoolkit=9.2 -c pytorch

这一步骤会自动处理所有必要的依赖关系，并设置好适合当前系统的 PyTorch 版本[^2]。

#### 验证安装是否成功
完成上述操作之后，可以通过运行简单的测试脚本来验证 PyTorch 是否能够识别到 GPU 设备。打开 Python 解释器或 Jupyter Notebook 并输入以下代码片段来进行检测：

import torch
print(torch.cuda.is_available())
print(torch.__version__)

如果一切正常，`torch.cuda.is_available()` 应该返回 `True` 表明 GPU 加速功能可用；同时也会打印出已安装的 PyTorch 版本号。