：PyTorch的Ubuntu之旅

PyTorch， 这玩意儿深厚度学界的明星框架，让许许多开发者为之倾倒。而Ubuntu，作为Linux系统中的佼佼者，天然成为了PyTorch的忠实拥趸。那么在Ubuntu上安装PyTorch，都需要准备哪些依赖项呢？让我们一起揭开这神秘的面纱。

一、 系统周围准备

在开头安装PyTorch之前，先说说需要确保你的Ubuntu系统满足以下要求：

• 操作系统：Ubuntu 18.04或更高大版本
• Python版本：Python 3.6或更高大版本
• CUDA版本：根据你的GPU型号选择合适的CUDA版本

二、依赖项安装

在Ubuntu上安装PyTorch，需要安装以下依赖项：

1. pip

pip是Python的包管理工具，用于安装和管理Python包。在安装PyTorch之前，请确保已安装pip。如果没有安装， 能用以下命令安装：

sudo apt install python3-pip

2. CUDA

如果你打算利用GPU进行加速，觉得能安装相应版本的CUDA。你能通过以下命令检查系统中是不是已安装CUDA：

nvidia-smi

如果未安装CUDA，请访问NVIDIA官网下载并安装适合你系统的CUDA版本。

3. cuDNN

cuDNN是用于深厚度神经网络的GPU加速库。你需要根据安装的CUDA版本安装相应的cuDNN版本。比方说如果你安装了CUDA 11.1，能从NVIDIA官网下载并安装cuDNN 8.0.5。

4. 其他依赖项

根据不同的安装方法和系统配置，兴许还需要安装其他依赖项。

sudo apt-get install -y build-essential curl git-core libssl-dev zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget libffi-dev llvm xz-utils libncurses5-dev tk-dev libxmlsec1-dev libxml2-dev

三、 PyTorch安装

在安装PyTorch之前，请确保你的Python周围已经准备优良。

1. 用pip安装

在终端中运行以下命令安装PyTorch：

pip install torch torchvision torchaudio

2. 用conda安装

如果你用conda进行Python包管理， 能用以下命令安装PyTorch：

conda install pytorch torchvision torchaudio cudatoolkit=11.6 -c pytorch -c conda-forge

在Ubuntu上安装PyTorch，需要准备一系列依赖项，包括pip、CUDA、cuDNN等。通过以上步骤，你能在Ubuntu上顺利安装PyTorch，开启你的深厚度学之旅。

---