1 环境
CentOS
2 查看显卡安装
lspci | grep -i nvidia
3b:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
5e:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
86:00.0 3D controller: NVIDIA Corporation GP104GL [Tesla P4] (rev a1)
3 安装驱动程序
3.1 下载驱动
传送门:https://www.nvidia.cn/Download/index.aspx?lang=cn
选择对应的显卡版本.
3.2 安装驱动
下载文件:NVIDIA-Linux-x86_64-418.67.run
3.2.1 准备操作
- 修改权限
添加可运行.
sudo chomd +x NVIDIA-Linux-x86_64-418.67.run
- 退出GUI界面
退出X Server,才能安装驱动.
sudo systemctl stop gdm.service
# 停止
sudo systemctl stop gdm.service
# 查看状态
sudo systemctl status gdm.service
# 启动
sudo systemctl start gdm.service
- 关闭Nouveau
cd /usr/lib/modprobe.d
sudo vim dist-blacklist.conf
blacklist nouveau
options nouveau modeset=0
- 查看nouveau状态
lsmod | grep nouveau
- 备份镜像
sudo mv /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r)-nouveau.img
- 新建镜像
sudo dracut /boot/initramfs-$(uname -r).img $(uname -r)
- 内核版本
cd /usr/src/kernels
ls
- 完整内核版本
3.10.0-957.27.2.el7.x86_64
使用uname -r
获取的内核版本不全.
3.2.2 安装插件
- 安装gcc
编译C.
sudo yum install -y gcc
- 安装gcc++
编译C++
sudo yum install -y gcc-c++
- 安装kernel-devel
内核依赖
sudo yum install -y kernel-devel
- 安装kernel-headers
内核头文件
sudo yum install -y kernel-headers
- 安装驱动
sudo ./NVIDIA-Linux-x86_64-418.67.run --no-opengl-files --kernel-source-path=/usr/src/kernels/3.10.0-957.27.2.el7.x86_64
使用root安装,不安装opengl,因为安装opengl,CentOS界面UI不能正常启动,添加内核路径.
3.2.3 查看显卡信息
nvidia-smi
4 安装CUDA
4.1 下载
传送门:https://developer.nvidia.com/cuda-toolkit-archive
CUDA8.0
4.2 安装
- 添加可执行权限
sudo chmod +x cuda_8.0.44_linux.run
- 安装CUDA
sudo ./cuda_8.0.44_linux.run --no-opengl-libs
- 跳过阅读许可
q
- 同意
accept
- 不安装Nvidia加速
这个加速驱动是旧版本,已经安装了最新的驱动,所以这个不安装,选择n
.
Install NVIDIA Accelerated Graphics Driver for Linux-x86_64 367.48?
(y)es/(n)o/(q)uit: n
- 安装Toolkit
Install the CUDA 8.0 Toolkit?
(y)es/(n)o/(q)uit: y
- 默认安装目录
Enter Toolkit Location
[ default is /usr/local/cuda-8.0 ]:
- 安装symbolic
Do you want to install a symbolic link at /usr/local/cuda?
(y)es/(n)o/(q)uit: y
- 卸载CUDA
yum uninstall /usr/local/cuda-8.0/bin
4.3 配置路径
- 配置
sudo vim ~/.bashrc
export CUDA_HOME=/usr/local/cuda-8.0
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64
- 生效
source ~/.bashrc
4.4 查看CUDA版本
- 详细信息
nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2016 NVIDIA Corporation
Built on Sun_Sep__4_22:14:01_CDT_2016
Cuda compilation tools, release 8.0, V8.0.44
- 版本信息
cat /usr/local/cuda/version.txt
CUDA Version 8.0.44
5 安装cuDNN
5.1 下载
传送门:https://developer.nvidia.com/rdp/cudnn-archive
与CUDA版本对应:cuDNN6.0
5.2 解压
CUDA安装目录:/usr/local/cuda
和/usr/local/cuda-8.0
tar -zxvf cudnn-8.0-linux-x64-v6.0.tgz -C /usr/local
直接将添加的文件解压到local目录下的cuda和cuda-8.0文件夹.
5.3 查看cuDNN信息
cat /usr/local/cuda-8.0/include/cudnn.h | grep CUDNN_MAJOR -A 2
#define CUDNN_MAJOR 6
#define CUDNN_MINOR 0
#define CUDNN_PATCHLEVEL 21
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)
#include "driver_types.h"
[参考文献]
[1]https://www.jianshu.com/p/a201b91b3d96
[2]https://blog.csdn.net/qq_41493990/article/details/82183035
[3]https://blog.csdn.net/fortuna_i/article/details/83141070
[4]https://www.jianshu.com/p/36dcf5185f01
[5]https://blog.csdn.net/zhe_csdn/article/details/88427829
[6]https://www.linuxidc.com/Linux/2017-01/139319.htm
[7]https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html