PyTorch在Ubuntu上的安装教程是什么

在Ubuntu上安装PyTorch可以分为几个步骤，具体如下：

安装前准备

1. 安装Python和pip：

   sudo apt-get update
   sudo apt-get install python3 python3-pip
   

2. 安装CUDA和cuDNN（如果使用GPU版本）：

   • 访问NVIDIA官网下载并安装适合Ubuntu版本的CUDA Toolkit。
   • 下载并安装cuDNN。

3. 创建虚拟环境（推荐）：

   python3 -m venv pytorch_env
   source pytorch_env/bin/activate
   

安装PyTorch

1. 安装CPU版本：

   pip3 install torch torchvision
   

2. 安装GPU版本：

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

   将替换为你安装的CUDA版本号，例如11.4。

验证安装

在Python交互环境中导入PyTorch并验证安装是否成功：

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

如果torch.cuda.is_available()返回True，则表示CUDA可用，你正在使用GPU版本的PyTorch。