温馨提示×

Linux PyTorch如何进行版本更新

小樊
73
2025-08-21 18:39:13
栏目: 智能运维

在Linux下更新PyTorch版本,可使用pip或conda工具,具体如下:

  • pip更新
    1. 打开终端,检查pip是否安装,未安装则用sudo apt-get install python3-pip安装。
    2. 升级pip到最新版:pip install --upgrade pip
    3. 更新PyTorch及其相关库:pip install --upgrade torch torchvision torchaudio。若需指定版本,可在命令后添加对应版本号,如pip install torch==2.3.0 torchvision==0.18.0 torchaudio==2.3.0
  • conda更新
    1. 打开终端,更新conda:conda update conda
    2. 查看当前PyTorch版本:conda list torch
    3. 安装或更新PyTorch,根据CUDA版本选择命令,如CUDA 12.1版本:conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia;CPU版本:conda install pytorch torchvision torchaudio cpuonly -c pytorch

更新前建议备份项目代码和数据,更新后可通过import torch; print(torch.__version__)验证安装是否成功。

0