欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页

安装NVIDIA显卡驱动 Cuda Cudnn

程序员文章站 2022-04-08 08:51:57
...

环境:ubuntu16.04

安装NVIDIA显卡驱动

禁用旧的显卡驱动
按照这篇blog ubuntu禁用nouveau方法,成功禁用旧的显卡驱动。之后可执行 lsmod | grep nouveau 命令查看nouveau驱动是否完全被禁用,无结果显示已禁用。
之前按照学长blog安装NVIDIA显卡驱动,一重启就进入循环登录界面,于是按照这篇blogUbuntu 16.04 用户登录界面死循环问题的解决重新安装,NVIDIA-Linux-x86_64-381.22.run安装包可在官网下载。

# 进入文本模式
Ctrl+Alt+F1
# 关闭lightdm服务
sudo service lightdm stop
cd Downloads
sudo chmod +x NVIDIA*.run
# 驱动安装
sudo ./NVIDIA-Linux-x86_64-381.22.run -no-x-check -no-nouveau-check -no-opengl-files
# 回到图形界面
sudo service lightdm restart
# 重启
reboot

驱动安装好后,重启依旧循环登录,于是只好换一种安装方式:

sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt-get update
sudo apt-get install nvidia-384

重启,终于安装成功。

安装NVIDIA显卡驱动 Cuda Cudnn

安装Cuda

download cuda_8.0.61_375.26_linux.run文件成功安装:

sudo bash cuda_8.0.61_375.26_linux.run
# /usr/local中能看到cuda,cuda-9.1的库
# 配置环境变量

安装NVIDIA显卡驱动 Cuda Cudnn

安装NVIDIA显卡驱动 Cuda Cudnn

安装NVIDIA显卡驱动 Cuda Cudnn

cudnn

tar zxvf cudnn-8.0-linux-x64-v5.1.tgz

janie@B250M-D3H:~/Downloads/cuda/lib64$ sudo cp lib* /usr/local/cuda/lib64/
janie@B250M-D3H:~/Downloads/cuda/lib64$ cd ..
janie@B250M-D3H:~/Downloads/cuda$ cd include/
janie@B250M-D3H:~/Downloads/cuda/include$ sudo cp cudnn.h /usr/local/cuda/include/

janie@B250M-D3H:~/Downloads/cuda/include$ cd /usr/local/cuda/lib64/
janie@B250M-D3H:/usr/local/cuda/lib64$ sudo rm -rf libcudnn.so libcudnn.so.5
janie@B250M-D3H:/usr/local/cuda/lib64$ sudo ln -s libcudnn.so.5.1.5 libcudnn.so.5
janie@B250M-D3H:/usr/local/cuda/lib64$ sudo ln -s libcudnn.so.5 libcudnn.so

janie@B250M-D3H:/usr/local/cuda/lib64$ sudo ldconfig

安装NVIDIA显卡驱动 Cuda Cudnn