debian

Debian如何管理多个PyTorch版本

小樊
40
2025-09-13 11:58:34
栏目: 智能运维

在Debian上管理多个PyTorch版本,推荐使用虚拟环境工具,如venvconda,以下是具体方法:

使用venv(Python自带)

  1. 安装venv(若未安装):sudo apt install python3-venv
  2. 为每个PyTorch版本创建虚拟环境,如python3 -m venv pytorch1.7
  3. 激活虚拟环境:source pytorch1.7/bin/activate
  4. 安装对应版本的PyTorch,如pip install torch==1.7.0 torchvision==0.8.1
  5. 验证安装:import torch; print(torch.__version__)
  6. 退出虚拟环境:deactivate

使用conda(需先安装Anaconda/Miniconda)

  1. 创建新的Conda环境,如conda create -n pytorch1.8 python=3.8
  2. 激活环境:conda activate pytorch1.8
  3. 安装特定版本的PyTorch,如conda install pytorch=1.8.0 torchvision=0.9.0 -c pytorch
  4. 验证安装:import torch; print(torch.__version__)
  5. 退出环境:conda deactivate

0
看了该问题的人还看了