Ubuntu系统中如何卸载PyTorch

在Ubuntu系统中卸载PyTorch，可以按照以下步骤进行：

方法一：使用pip卸载

1. 打开终端：

   • 可以通过快捷键 Ctrl + Alt + T 打开终端。

2. 卸载PyTorch：

   • 如果你是通过pip安装的PyTorch，可以使用以下命令卸载：

     pip uninstall torch torchvision torchaudio
     

   • 如果你有多个Python版本，确保使用正确的pip版本（例如，对于Python 3.x，使用 pip3）：

     pip3 uninstall torch torchvision torchaudio
     

3. 验证卸载：

   • 卸载完成后，可以通过以下命令检查是否成功卸载：

     pip show torch
     

   • 如果没有输出或者显示“Package not found”，则表示卸载成功。

方法二：使用conda卸载（如果你是通过conda安装的）

1. 打开终端：

   • 同样使用 Ctrl + Alt + T 打开终端。

2. 卸载PyTorch：

   • 使用conda命令卸载PyTorch：

     conda remove pytorch torchvision torchaudio cudatoolkit=11.3 -y
     

   • 注意：cudatoolkit=11.3 是一个示例版本号，你需要根据你安装时使用的CUDA版本进行调整。如果不使用GPU，可以省略 -cudatoolkit=11.3 部分。

3. 验证卸载：

   • 卸载完成后，可以通过以下命令检查是否成功卸载：

     conda list pytorch
     

   • 如果没有输出或者显示“pytorch not found”，则表示卸载成功。

清理残留文件

有时候，即使卸载了PyTorch，一些残留文件可能仍然存在。你可以手动删除这些文件：

1. 删除Python包目录：

   • 默认情况下，Python包安装在 ~/.local/lib/pythonX.Y/site-packages/ 目录下。你可以删除与PyTorch相关的文件夹：

     rm -rf ~/.local/lib/python3.x/site-packages/torch*
     rm -rf ~/.local/lib/python3.x/site-packages/vision*
     rm -rf ~/.local/lib/python3.x/site-packages/audio*
     

2. 删除缓存文件：

   • 你还可以删除pip或conda的缓存文件：

     pip cache purge
     conda clean --all
     

通过以上步骤，你应该能够成功地在Ubuntu系统中卸载PyTorch及其相关组件。