Ubuntu如何安装PyTorch

在Ubuntu上安装PyTorch，你可以按照以下步骤进行：

方法一：使用pip安装

1. 更新系统包：

   sudo apt update
   sudo apt upgrade
   

2. 安装Python和pip（如果尚未安装）：

   sudo apt install python3 python3-pip
   

3. 安装PyTorch： 访问PyTorch官网，选择适合你系统的安装命令。以下是一些常见的安装命令示例：

   • CPU版本：

     pip3 install torch torchvision torchaudio
     

   • CUDA 11.3版本（如果你有NVIDIA GPU并且已经安装了CUDA 11.3）：

     pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
     

   • CUDA 11.7版本：

     pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117
     

方法二：使用conda安装

如果你更喜欢使用Anaconda或Miniconda，可以通过conda来安装PyTorch。

1. 安装Anaconda或Miniconda（如果尚未安装）：

   • 下载并安装Anaconda或Miniconda。

2. 创建并激活conda环境（可选）：

   conda create -n pytorch_env python=3.8
   conda activate pytorch_env
   

3. 安装PyTorch： 访问PyTorch官网，选择适合你系统的安装命令。以下是一些常见的安装命令示例：

   • CPU版本：

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • CUDA 11.3版本（如果你有NVIDIA GPU并且已经安装了CUDA 11.3）：

     conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
     

   • CUDA 11.7版本：

     conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch
     

验证安装

安装完成后，你可以通过以下命令验证PyTorch是否安装成功：

import torch
print(torch.__version__)
print(torch.cuda.is_available())  # 如果安装了CUDA版本，应该返回True

如果一切正常，你应该能够看到PyTorch的版本号，并且如果安装了CUDA版本，torch.cuda.is_available()应该返回True。

希望这些步骤能帮助你在Ubuntu上成功安装PyTorch！