在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及其依賴被徹底清除。