温馨提示×

在Ubuntu上如何更新PyTorch版本

小樊
42
2025-08-13 22:42:26
栏目: 智能运维

在Ubuntu上更新PyTorch版本,常用方法有以下两种:

  • 使用pip安装
    1. 打开终端,通过pip show torch查看当前版本。
    2. 升级pip:pip install --upgrade pip
    3. 卸载旧版本:pip uninstall torch torchvision torchaudio
    4. 安装最新版:pip install torch torchvision torchaudio。若需特定版本,可指定版本号,如pip install torch==<version> torchvision==<version> torchaudio==<version>
    5. 验证更新:pip show torch查看版本信息。
  • 使用conda安装
    1. 打开终端,激活conda环境:conda activate <your_env>
    2. 检查当前版本:conda list torch
    3. 卸载旧版本:conda remove pytorch torchvision torchaudio
    4. 安装最新版:conda update pytorch torchvision torchaudio cudatoolkit=<your_cuda_version> -c pytorch,其中<your_cuda_version>替换为实际CUDA版本,若不用GPU可省略相关参数。
    5. 验证更新:conda list torch查看版本。

0