Ubuntu中PyTorch依赖怎么装

在Ubuntu系统中安装PyTorch及其依赖，可以按照以下步骤进行：

方法一：使用pip安装

1. 更新系统包列表

   sudo apt update
   

2. 安装Python和pip 如果你还没有安装Python和pip，可以使用以下命令：

   sudo apt install python3 python3-pip
   

3. 安装PyTorch PyTorch官方提供了详细的安装指南，你可以根据你的CUDA版本选择合适的命令。以下是一些常见CUDA版本的示例：

   • 无GPU支持（CPU版本）

     pip3 install torch torchvision torchaudio
     

   • 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
     

   • CUDA 10.2支持

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

   你可以访问PyTorch官网获取最新的安装命令。

方法二：使用conda安装

如果你更喜欢使用conda来管理包和环境，可以按照以下步骤操作：

1. 安装Miniconda或Anaconda 如果你还没有安装conda，可以从Miniconda官网或Anaconda官网下载并安装。

2. 创建新的conda环境

   conda create -n pytorch_env python=3.8
   

3. 激活环境

   conda activate pytorch_env
   

4. 安装PyTorch 同样，你可以根据你的CUDA版本选择合适的命令：

   • 无GPU支持（CPU版本）

     conda install pytorch torchvision torchaudio cpuonly -c pytorch
     

   • 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
     

   • CUDA 10.2支持

     conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
     

验证安装

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

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

希望这些步骤能帮助你在Ubuntu系统中成功安装PyTorch及其依赖！如果有任何问题，请随时提问。