阅读量:2
在Ubuntu上更新PyTorch版本,常用方法如下:
- 使用pip安装:
- 卸载旧版本:
pip uninstall torch torchvision torchaudio。 - 安装最新版:
pip install --upgrade torch torchvision torchaudio。若需特定版本,可指定版本号,如pip install torch==。若用GPU,需指定CUDA版本,如torchvision== torchaudio== pip install torch torchvision torchaudio --extra -index -url https://download.pytorch.org/whl/cu117(以CUDA 11.7为例)。
- 卸载旧版本:
- 使用conda安装:
- 激活conda环境(若有):
conda activate your_environment_name。 - 卸载旧版本:
conda remove pytorch torchvision torchaudio。 - 安装最新版:
conda update pytorch torchvision torchaudio cudatoolkit=<对应版本> -c pytorch。不使用GPU可省略cudatoolkit参数。
- 激活conda环境(若有):
安装完成后,可通过import torch; print(torch.__version__)验证版本。操作前建议备份代码数据,确保CUDA等版本兼容,且在虚拟环境中操作。
以上就是关于“如何更新Ubuntu上的PyTorch版本”的相关介绍,筋斗云是国内较早的云主机应用的服务商,拥有10余年行业经验,提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩,主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复;电子化备案,提交快速,专业团队7×24小时服务支持!
简单好用、高性价比云服务器租用链接:https://www.jindouyun.cn/product/cvm