温馨提示×

Ubuntu PyTorch如何卸载干净

小樊
71
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