如何安装 PyTorch

PyTorch 是一个基于 Python 的科学计算包，主要针对两类人群：

• 作为 NumPy 的替代品，利用 GPU 的性能进行计算；
• 作为深度学习研究平台，提供了最大的灵活性和速度。

本文将介绍如何在 Shell-Bash 中安装 PyTorch。

安装步骤

以下是在 Shell-Bash 中安装 PyTorch 的详细步骤：

1. 安装 Anaconda

前往 Anaconda 的官网，根据自己系统的版本下载安装包进行安装。

2. 创建虚拟环境

打开终端（Terminal），输入以下命令创建一个名为“pytorch”的虚拟环境（可以自定义名称）：

$ conda create -n pytorch python=3.8

在此过程中，会提示询问是否安装虚拟环境，输入y进行确认。

3. 激活虚拟环境

激活虚拟环境，使得在该环境下安装的包只对该环境生效：

$ conda activate pytorch

4. 安装 PyTorch

在激活的虚拟环境中，输入以下命令安装 PyTorch：

$ conda install pytorch torchvision torchaudio cpuonly -c pytorch

如果希望安装 GPU 版本的 PyTorch，请在以上命令中删除 -cpuonly 安装参数。

5. 验证安装是否成功

以下是验证 PyTorch 是否成功安装的代码片段：

# 导入 PyTorch 库
import torch

# 创建张量
x = torch.rand(5, 3)

# 打印张量
print(x)

如果输出了一个形状为 (5, 3) 的随机张量，则表明安装成功。

总结

本文介绍了在 Shell-Bash 中安装 PyTorch 的详细过程，希望对初学者有所帮助。在此过程中，需要注意 Anaconda 的安装和虚拟环境的激活，以及根据自己的需要选择 CPU 或 GPU 版本的 PyTorch 进行安装。