在Debian上管理多个PyTorch版本,推荐使用虚拟环境工具,如venv
或conda
,以下是具体方法:
venv
(若未安装):sudo apt install python3-venv
。python3 -m venv pytorch1.7
。source pytorch1.7/bin/activate
。pip install torch==1.7.0 torchvision==0.8.1
。import torch; print(torch.__version__)
。deactivate
。conda create -n pytorch1.8 python=3.8
。conda activate pytorch1.8
。conda install pytorch=1.8.0 torchvision=0.9.0 -c pytorch
。import torch; print(torch.__version__)
。conda deactivate
。