在Ubuntu上彻底卸载PyTorch需根据安装方式选择对应方法,并清理残留文件,具体步骤如下:
pip安装
pip uninstall torch torchvision torchaudio
(若使用Python 3且系统有Python 2,需用pip3)
conda安装
conda remove pytorch torchvision torchaudio -c pytorch --yes
(--yes跳过确认,-c pytorch指定频道)
源码编译安装
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" # 若报错则已卸载
sudo apt-get --purge remove cuda # 谨慎操作,可能影响其他程序
按以上步骤可确保PyTorch及其依赖被彻底清除。