温馨提示×

Ubuntu下PyTorch库更新方法

小樊
52
2025-09-13 01:39:21
栏目: 智能运维

在Ubuntu下更新PyTorch库,主要有以下两种方法:

  • 使用pip更新
    1. 打开终端(Ctrl+Alt+T)。
    2. 升级pip(可选):pip install --upgrade pip
    3. 卸载旧版本(可选):pip uninstall torch torchvision torchaudio
    4. 安装最新版:pip install --upgrade torch torchvision torchaudio。若需特定版本,可指定版本号,如pip install torch==<version> torchvision==<version> torchaudio==<version>
  • 使用conda更新
    1. 打开终端。
    2. 激活conda环境(若有):conda activate your_environment_name
    3. 更新conda(可选):conda update conda
    4. 卸载旧版本(可选):conda remove pytorch torchvision torchaudio cudatoolkit
    5. 安装新版本:conda install pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch,不使用GPU可省略cudatoolkit参数。

更新前建议备份项目文件和数据,确保CUDA等版本兼容,若在虚拟环境中操作需先激活环境。

0