阅读量:3
在Ubuntu中更新PyTorch版本,常用方法有pip和conda两种,具体如下:
- pip安装:
- 卸载旧版本:
pip uninstall torch torchvision torchaudio。 - 访问PyTorch官网,根据系统、CUDA版本等选择安装命令。如CUDA 11.7版本:
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu117;CPU版本:pip install torch torchvision torchaudio。 - 验证安装:
import torch,print(torch.__version__)。
- 卸载旧版本:
- conda安装:
- 卸载旧版本:
conda remove pytorch torchvision torchaudio。 - 访问PyTorch官网,根据系统、CUDA版本等选择安装命令。如CUDA 11.7版本:
conda install pytorch torchvision torchaudio cudatoolkit=11.7 -c pytorch;CPU版本:conda install pytorch torchvision torchaudio cpuonly -c pytorch。 - 验证安装:
conda list torch查看版本信息。
- 卸载旧版本:
以上就是关于“Ubuntu中如何更新PyTorch版本”的相关介绍,筋斗云是国内较早的云主机应用的服务商,拥有10余年行业经验,提供丰富的云服务器、租用服务器等相关产品服务。云服务器资源弹性伸缩,主机vCPU、内存性能强悍、超高I/O速度、故障秒级恢复;电子化备案,提交快速,专业团队7×24小时服务支持!
简单好用、高性价比云服务器租用链接:https://www.jindouyun.cn/product/cvm