阅读量:3
在Ubuntu上管理PyTorch版本，可通过以下方式操作：
一、安装/更新PyTorch
-
pip方式
- 卸载旧版本：
pip uninstall torch torchvision torchaudio - 安装指定版本（需搭配CUDA版本）：
- GPU版本：
pip install torch==2.0.1+cu118 --extra-index-url https://download.pytorch.org/whl/cu118 - CPU版本：
pip install torch torchvision torchaudio
- GPU版本：
- 卸载旧版本：
-
conda方式
- 卸载旧版本：
conda remove pytorch torchvision torchaudio - 安装指定版本：
- GPU版本：
conda install pytorch==2.0.1 torchvision torchaudio pytorch-cuda=11.8 -c pytorch -c nvidia - CPU版本：
conda install pytorch==2.0.1 torchvision torchaudio cpuonly -c pytorch
- GPU版本：
- 卸载旧版本：
二、版本切换与验证
-
查看当前版本：
import torch print(torch.__version__) # 查看PyTorch版本 print(torch.version.cuda) # 查看CUDA版本（若安装GPU版） -
切换虚拟环境：
- 激活环境：
source venv_name/bin/activate（venv）或conda activate env_name（conda） - 重新安装对应环境的PyTorch版本即可完成切换。
- 激活环境：
三、注意事项
- CUDA兼容性：需确保PyTorch版本与CUDA版本匹配（如PyTorch 2.0.1需CUDA 11.8）。
- 环境隔离：建议使用虚拟环境（venv/conda）避免版本冲突。
- 驱动检查：通过
nvidia-smi确认GPU驱动支持安装的CUDA版本。
参考来源：
以上就是关于“PyTorch在Ubuntu上如何进行版本管理”的相关介绍，筋斗云是国内较早的云主机应用的服务商，拥有10余年行业经验，提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩，主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复；电子化备案，提交快速，专业团队7×24小时服务支持！
简单好用、高性价比云服务器租用链接：https://www.jindouyun.cn/product/cvm