温馨提示×

Linux PyTorch版本如何更新

小樊
61
2025-09-04 19:30:09
栏目: 智能运维

在Linux系统下,可通过pip或conda两种方式更新PyTorch版本,具体如下:

  • 使用pip更新
    1. 打开终端,通过pip --version检查pip是否安装,未安装则用sudo apt-get install python3-pip安装。
    2. pip install --upgrade pip将pip升级到最新版本。
    3. 执行pip install --upgrade torch torchvision torchaudio更新PyTorch及其相关库到最新版。若需指定版本,如pip install torch==2.8.0 torchvision==0.22.0 torchaudio==2.8.0
  • 使用conda更新
    1. 打开终端,用conda update conda将conda更新到最新版本。
    2. 激活conda环境(若有):conda activate your_environment_name
    3. 执行conda update pytorch torchvision torchaudio cudatoolkit=版本号 -c pytorch,版本号根据CUDA版本选择,不使用GPU可省略cudatoolkit参数。

0