这几天在跑深度学习有关的代码,弄了一个GPU,找了很多博客去了解如何使用配置才可以使我的代码可以使用GPU去跑。在这一天的忙碌中,终于在晚上把这事搞定了。或许大家会疑惑一个"简单"配置为什么要搞这么久,-_-|| 因为大多资料都是关于Ubuntu系统的配置,关于Centos的资料挺少的,为此,特地记录下本次的配置过程与踩过的坑,留给大家参考。话不多说,开始本教程吧!
1.系统环境与软件版本
系统版本:cento7.5
内核版本:3.10.0-862.el7.x86_64
软件版本:cuda9.0与cudnn7.0
显卡驱动:Nvidia-384.183
cuda9.0具体包名:cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm
cudnn7.0具体包名:cudnn-9.0-linux-x64-v7.tgz
要注意版本的对应性,此处cuda9.0,所以cudnn选择的是cudnn7.0 for cuda9.0
即cudnn-9.0-linux-x64-v7.tgz
。
2.相应软件包查询与获取
2.1 查看当前系统版本:
系统中输入指令cat /etc/redhat-release
,结果图示例:
2.2 查看内核版本
输入指令uname -r
,结果图示例:
可以看到我们系统当前的内核版本,如我的为3.10.0-862.el7.x86_64
2.3 检测是否含有GPU
输入指令lspci | grep -i nvidia
,结果示例图:
可以看到我们系统是否含有GPU,其中图中的Tesla P40是我的GPU对应的版本;
2.4 获取对应版本的显卡驱动
在知道我们的GPU型号后可以去以下网址拿到对获取对应的显卡驱动:点此查询
界面如下图所示:
图中主要是选择我们的GPU类型,比如我的是Tesla
的,P系列的Tesla P40
的GPU,上图忘记选择cuda9.0
了,大家注意选择哈!因为tensorflow是基于cuda9.0的,貌似不支持cuda10(参考他人博客的,有误望指出)
点击右下方的SEARCH
,跳转到如下页面后,点击DOWNLOAD下载即可。
使用xftp将显卡驱动上传到我们的服务器中。
2.5 获取cuda9.0
下载地址:点此下载
按自己系统版本进行选择,选择示例如下图:
下载完后同样用xftp上传到我们的服务器中。
2.6 获取cudnn7.0
第一种方法:使用压缩包安装
在服务器中输入如下指令,下载对应的压缩包(即cuda9.0对应的cudnn7.0)
wget http://developer.download.nvidia.com/compute/redist/cudnn/v7.0.5/cudnn-9.0-linux-x64-v7.tgz
下载结果示例图:
更多类型的cudnn系列下载地址参考此处:点此查看
第二种方法:
注意:此处我亲测的是第一种方法的安装法师,第二种方法的话需要大家尝试哈!此处写出来是为了提供该官网下载地址。去NVIDIA官网下载cuda7.0,需要注册账号登录后才可下载:点此下载
3. 开始安装
3.1 安装前准备工作
yum安装epel、aliyun、elrepo源
yum -y install https://dl.fedoraproject.org/pub/epel/epel-release-latest-7.noarch.rpm
yum -y install epel-release
wget -O /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo
rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
rpm -Uvh https://www.elrepo.org/elrepo-release-7.0-3.el7.elrepo.noarch.rpm
安装gcc、gcc-c++编译器
若已存在则不需要安装
yum –y install gcc
yum -y install gcc-c++
禁用系统自带的nouveau驱动
1)编辑文件vi /etc/default/grub
在其中添加rd.driver.blacklist=nouveau nouveau.modeset=0
到GRUB_CMDLINE_LINUX
后面,如下图:
2)输入以下指令生成新的grub配置
grub2-mkconfig -o /boot/grub2/grub.cfg
3)编辑/etc/modprobe.d/blacklist.conf(如果不存在则创建),并添加
blacklist nouveau
vi /etc/modprobe.d/blacklist.conf
示例图:
检查内核版本是否一致
这一步如果不做好的话(即内核版本不一致),在安装显卡驱动时就会遇到下图类似的错误:
为此,我们需要检查一下当前运行内核版本与/usr/src/kernels
目录下的kernel源码版本是否一致。
- 检查当前运行内核版本
uname -r
,结果示例如下图:
- 检查
/usr/src/kernels
目录下的kernel源码版本
cd /usr/src/kernels
ls
可以看到有对应的版本3.10.0-862.el7.x86_64
。其实我之前就踩了这个坑,该文件夹下并没有3.10.0-862.el7.x86_64
,只有3.10.0-957.21.3.el7.x86_64
版本的源码,导致我装显卡驱动一致失败。
若版本一致,则可以继续进行下一步了(忽略下面版本不一致时候的操作)。
若无对应版本的内核源码,可以在此处下载:点此下载
找到与自己系统运行源码版本一致的包,如我的查找的包为:
使用xftp将该包传到服务器中,并cd到存储该包的文件夹中,执行以下指令:
yum -y install kernel-3.10.0-862.14.4.el7.x86_64.rpm
cd /usr/src/kernels
并查看文件夹是否安装成功。
重启使刚才的禁用配置生效
reboot
3.2 安装cuda9.0
cd到我们存放cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm
的文件夹中,执行安装命令:
rpm -i cuda-repo-rhel7-9-0-local-9.0.176-1.x86_64.rpm
yum clean all
yum -y install cuda
cuda的默认安装路径如下:/usr/local/cuda
设置cuda环境变量
vi /etc/profile
添加以下内容到末尾:
export CUDA_HOME=/usr/local/cuda
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$CUDA_HOME/extras/CUPTI/lib64:$LD_LIBRAY_PATH
export PATH=$CUDA_HOME/bin:$PATH
使配置文件生效:source /etc/profile
3.3 安装cudnn7.0
cd到我们的压缩包存放文件夹中,解压
tar -zxvf cudnn-9.0-linux-x64-v7.tgz
结果如图所示:
将cuda中的文件复制到cuda toolkit目录中,然后更改文件权限:
sudo cp cuda/include/cudnn.h /usr/local/cuda/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
至此cuda与cudnn安装完成。
3.4 安装tensorflow-gpu
yum -y install python-pip
pip install --upgrade pip
pip install tensorflow-gpu==1.10.1
之所以安装1.10.1是因为之前安装1.11.1时,导入tensorflow包使用时会产生段错误:
起初以为是cudnn版本问题,但实际是tensorflow-gpu版本过高了,降低为1.10.1就好了。
3.5 验证cuda和cudnn安装无误
进入到python shell,输入以下代码
import tensorflow
按回车,若无报错说明cuda与cudnn已经安装好了。
3.6 安装NVIDIA显卡驱动
重做inittramfs镜像:
cp /boot/initramfs-$(uname -r).img /boot/initramfs-$(uname -r).img.bak
dracut /boot/initramfs-$(uname -r).img $(uname -r)
cd到驱动所在目录:
chmod +x NVIDIA-Linux-x86_64-384.183.run
./NVIDIA-Linux-x86_64-384.183.run --kernel-source-path=/usr/src/kernels/3.10.0-862.el7.x86_64/
执行后就开始安装驱动了,
接受许可:
检查驱动安装情况nvidia-smi
,出现如下页面说明已经成功了。
至此,全部安装已经完成了!可以happy的使用GPU了!
参考
显卡驱动安装:https://blog.csdn.net/xueshengke/article/details/78134991
cudnn压缩包:https://blog.csdn.net/xiangxianghehe/article/details/79177833
导入tensorflow报错解决:https://blog.csdn.net/u014561933/article/details/80201552
官方安装文档:https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-linux
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html#install-linux
https://docs.nvidia.com/cuda/cuda-installation-guide-linux/index.html
cudnn报错解决:https://blog.csdn.net/jy1023408440/article/details/82887479