温馨提示×

Ubuntu PyTorch如何卸载干净

小樊
59
2025-08-19 22:55:19
栏目: 智能运维

在Ubuntu上彻底卸载PyTorch需根据安装方式选择对应方法,并清理残留文件,具体步骤如下:

一、根据安装方式卸载

  1. pip安装

    pip uninstall torch torchvision torchaudio 

    (若使用Python 3且系统有Python 2,需用pip3

  2. conda安装

    conda remove pytorch torchvision torchaudio -c pytorch --yes 

    --yes跳过确认,-c pytorch指定频道)

  3. 源码编译安装

    cd /path/to/pytorch # 进入源码目录  python setup.py clean rm -rf build/ /usr/local/lib/python*/dist-packages/torch* 

    (需手动删除安装目录及构建文件)

二、清理残留文件

  • 删除配置和缓存
    rm -rf ~/.torch/ ~/.cache/torch/ 
  • 删除系统级文件
    sudo rm -rf /usr/local/lib/python*/dist-packages/torch* 
    (需根据实际Python版本调整路径)

三、验证卸载

pip show torch # 若显示“not found”则卸载成功  conda list | grep torch # 应无输出  python -c "import torch" # 若报错则已卸载  

注意事项

  • 若曾安装CUDA相关组件,可额外执行:
    sudo apt-get --purge remove cuda # 谨慎操作,可能影响其他程序  
  • 操作前建议备份重要数据,避免误删。

按以上步骤可确保PyTorch及其依赖被彻底清除。

0