ubuntu18.04卸载cpu驱动pytorch
时间: 2024-12-04 07:22:41 浏览: 3
在Ubuntu 18.04上卸载CPU驱动与PyTorch关联的话,通常是针对CUDA版本不兼容或需要回滚到纯CPU支持的情况。以下是删除CPU驱动的一般步骤:
1. **确认安装了哪些cuda版本**:
打开终端并运行:
```
dpkg -l | grep nvidia
```
查看已安装的nvidia相关包。
2. **如果有的话,移除特定版本的CUDA库**:
如果你想卸载特定版本,例如cuda-10.0,可以这样做:
```
sudo apt-10-0 nvidia-cuda-dev-10-0
```
3. **卸载cuDNN**:
```
sudo apt-get remove libnvidia-cudnn*
```
4. **清除CUDA相关配置**:
```
rm /etc/apt/sources.list.d/cuda.list
```
5. **更新软件包列表并清理缓存**:
```
sudo apt-get update && sudo apt-get autoremove
```
6. **关闭或禁用GPU服务**:
```
sudo systemctl disable nvidia* && sudo systemctl stop nvidia*
```
7. **最后,如果有PyTorch的环境变量设置,也需要移除它们**:
清理pip环境:
```
pip uninstall torch torchvision -y
```
8. **检查是否成功**:
使用`dpkg -l | grep python-torch`检查是否有PyTorch的相关包还在系统里。
**注意**:操作前建议备份重要数据,因为上述命令可能会对系统产生影响。此外,如果你依赖的是默认的CPU驱动,无需特意卸载。
阅读全文