ubuntu18.04卸载cpu驱动pytorch

时间: 2024-12-04 08:22:41 浏览: 90
DOC

ubuntu18.04安装pytorch1.9.0和torchvision0.10.0

在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驱动,无需特意卸载。
阅读全文

相关推荐