Debian如何管理PyTorch的不同版本

2025年7月21日 互联网

在Debian系统上管理PyTorch的不同版本可以通过以下几种方法实现：

使用pip管理PyTorch版本

  1. 安装pip（如果尚未安装）：
sudo apt update
sudo apt install python3-pip

  1. 安装特定版本的PyTorch：

    • 无CUDA（CPU版本）： 
      pip3 install torch torchvision torchaudio
    • CUDA 10.2： 
      pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu102
    • 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 12.0： 
      pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu120

  2. 更新PyTorch：

    • 卸载旧版本： 
      pip3 uninstall torch torchvision torchaudio
    • 安装最新版本： 
      pip3 install torch torchvision torchaudio

  3. 验证安装：

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

使用conda管理PyTorch版本

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

    • 下载并运行安装脚本： 
      wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
bash Miniconda3-latest-Linux-x86_64.sh

  2. 创建并激活conda环境：

    conda create -n pytorch_env python=3.8
conda activate pytorch_env

  3. 安装特定版本的PyTorch：

    • 无CUDA（CPU版本）： 
      conda install pytorch torchvision torchaudio cpuonly -c pytorch
    • CUDA 10.2： 
      conda install pytorch torchvision torchaudio cudatoolkit=10.2 -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 12.0： 
      conda install pytorch torchvision torchaudio cudatoolkit=12.0 -c pytorch

  4. 更新PyTorch：

    • 卸载旧版本： 
      conda remove pytorch torchvision torchaudio cudatoolkit -n pytorch_env
    • 安装最新版本： 
      conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

  5. 验证安装：

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

使用虚拟环境管理不同版本的PyTorch

  1. 创建虚拟环境：

    python3 -m venv pytorch-env
source pytorch-env/bin/activate

  2. 在虚拟环境中安装特定版本的PyTorch：

    • 无CUDA（CPU版本）： 
      pip install torch torchvision torchaudio
    • CUDA 10.2： 
      pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu102

  3. 切换虚拟环境：

    source /path/to/other/pytorch-env/bin/activate

通过以上方法，你可以在Debian系统上灵活地管理PyTorch的不同版本，确保每个项目都能在隔离的环境中运行，避免版本冲突。

最新文章