Ubuntu卸载nvidia驱动并安装CUDA+CUDNN

1. 切换到集成显卡

下图第二项


Ubuntu卸载nvidia驱动并安装CUDA+CUDNN_第1张图片

2. 卸载之前的驱动

sudo apt-get --purge remove nvidia*

sudo apt autoremove

sudo apt-get --purge remove "*cublas*" "cuda*"

sudo apt-get --purge remove "*nvidia*"

3. 下载对应的cuda cudnn,并且看看显卡是否支持cuda

https://www.geforce.cn/hardware/technology/cuda/supported-gpus

4. 切换到tty2终端(ctrl + alt + f2)

不要安装nvidia驱动,【直接安装cuda 直接安装cuda 直接安装cuda】,貌似现在cuda有集成的nvidia驱动

5. 安装cudnn(根据自己的替换)并检测是否成功

tar -xvf cudnn-9.0-linux-x64-v7.tgz

cd cuda 

sudo cp include/cudnn.h /usr/local/cuda/include/ 

sudo cp lib64/*  /usr/local/cuda/lib64/

sudo chmod a+r /usr/local/cuda/include/cudnn.h

sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

cd /usr/local/cuda/lib64/

sudo rm -rf libcudnn.so libcudnn.so.7

sudo ln -s libcudnn.so.7.0.5 libcudnn.so.7

sudo ln -s libcudnn.so.7 libcudnn.so


检测


cp -r /usr/src/cudnn_samples_v7/ /home/wdong/

cd /home/wdong/cudnn_samples_v7/mnistCUDNN

makeclean &&make

./mnistCUDNN

OK啦

你可能感兴趣的:(Ubuntu卸载nvidia驱动并安装CUDA+CUDNN)