当前位置: 首页 > 文档资料 > Caffe 学习笔记 >

安装 cuda 遇到的问题

优质
小牛编辑
127浏览
2023-12-01

安装 cuda 遇到了很多问题,这里简单记一下

下载

版本

查看 cuda 版本:

cat /usr/local/cuda/version.txt

查看 cudnn 版本:

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

查看显卡信息:

lspci | grep VGA

卸载

关闭 x:

gdm类型的桌面系统
sudo /etc/init.d/gdm stop
sudo /etc/init.d/gdm status

light类型的桌面系统
sudo /etc/init.d/lightdm stop
sudo /etc/init.d/lightdm status

卸载驱动:

sudo apt-get purge nvidia-cuda*
sudo apt-get purge nvidia-*

使用

nvidia-smi 实时刷新并高亮显示状态

watch -n 1 -d nvidia-smi