Ubuntu 20.04下安装PyTorch：从CPU到GPU的全面指南

Ubuntu20.04安装pytorch
在Ubuntu 20.04上安装PyTorch是一个常见的需求，尤其是在机器学习和深度学习的应用中。本篇文章将指导你如何顺利地在Ubuntu 20.04上安装PyTorch。
首先，我们需要确认你的系统已经更新到最新状态，并且安装了必要的依赖项。打开终端，输入以下命令：

sudo apt update
sudo apt upgrade

接下来，我们需要安装Python和pip。你可以使用以下命令来安装最新版本的Python 3和pip：

sudo apt install python3 python3-pip

安装完成后，我们就可以使用pip来安装PyTorch。需要注意的是，PyTorch需要GPU支持，因此如果你没有NVIDIA显卡或者不想使用GPU，你可以选择安装CPU版本的PyTorch。以下是在Ubuntu 20.04上安装CPU版PyTorch的步骤：

pip3 install torch torchvision torchaudio

如果你想安装GPU版本的PyTorch，你需要先安装NVIDIA的CUDA工具包。你可以从NVIDIA的官方网站下载并安装适合你显卡的CUDA版本。安装完成后，你需要设置环境变量，以便让系统知道CUDA的位置。你可以使用以下命令来设置环境变量：

echo 'export PATH=/usr/local/cuda-11.0/bin:$PATH' >> ~/.bashrc
echo 'export LD_LIBRARY_PATH=/usr/local/cuda-11.0/lib64:$LD_LIBRARY_PATH' >> ~/.bashrc
source ~/.bashrc

接下来，你需要安装cuDNN库。你可以从NVIDIA的官方网站下载并安装适合你CUDA版本的cuDNN库。安装完成后，你需要设置环境变量，以便让系统知道cuDNN的位置。你可以使用以下命令来设置环境变量：

echo 'export CUDNN_LIB_DIR=/usr/lib/x86_64-linux-gnu/' >> ~/.bashrc
echo 'export CUDNN_INCLUDE_DIR=/usr/include/' >> ~/.bashrc
source ~/.bashrc

最后，你可以使用以下命令来安装GPU版的PyTorch：

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

这样就可以在Ubuntu 20.04上成功安装PyTorch了。你可以通过运行以下命令来验证安装是否成功：

import torch
print(torch.__version__)

如果一切顺利，你应该能够看到PyTorch的版本号输出在终端上。现在你可以开始使用PyTorch进行机器学习和深度学习开发了。