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

安装前准备

1. 安装Python和pip：

   sudo apt-get updatesudo apt-get install python3 python3-pip

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

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

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

   python3 -m venv pytorch_envsource 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。