安装 PyTorch - Shell/Bash

PyTorch是Facebook AI Research团队开源的一个深度学习工具，它的使用已经广泛应用于学术界和工业界。本文将介绍如何安装PyTorch。

环境要求

• Python >= 3.6
• CUDA >= 10.1 (如果需要使用GPU)
• cuDNN >= 7.6 (如果需要使用GPU)

获取安装命令

首先，需要访问官方网站来获取安装PyTorch的命令。在网站上选择自己的电脑平台以及需要的PyTorch版本，然后复制安装命令到终端中（Unix 或 Windows）。

例如，在安装PyTorch 1.9.0版本的情况下，对于Linux/MacOS系统，可以使用以下命令：

pip3 install torch torchvision torchaudio

对于Windows系统，可以使用以下命令：

pip install torch torchvision torchaudio

注意：如果你的电脑环境中同时存在Python 2和Python 3，为了避免出错，应该使用pip3命令而不是pip命令。

安装过程

安装过程会在终端中显示，显示过程中可以看到哪些包被安装了。需要等待安装完成后，输出的最后一行应该是 "Successfully installed PyTorch"。

在 ~/.bashrc 或 ~/.bash_profile 中更新路径（如果需要），在这里更新可以让你在终端中使用PyTorch的命令更方便。

$ echo "export PATH=\$PATH:\$HOME/.local/bin" >> ~/.bashrc
$ source ~/.bashrc

验证安装

在终端中输入以下命令，验证是否安装成功：

python3 -c "import torch; print(torch.__version__); print(torch.cuda.is_available())"

输出应该如下：

1.9.0
True

总结

在本文中，我们介绍了如何安装PyTorch，并在结束前对安装过程进行了验证。希望这篇文章对你安装PyTorch有所帮助！