linux(centos)安装NVIDIA显卡驱动,安装CUDA,安装tensorflow

装系统的时候,各种对应的版本需要对上,不然中间会有各种坑,系统崩溃等等。ubunt装了几次各种问题,最后用centos 顺利的装好了。

写文章的时候 tensorflow只支持CUDA9.0, cuDNN v7.0, 所以需要装CUDA9.0,不能直接装最新版。(CUDA9.0支持 ubutnu17.04, 不支持最新的Ubuntu 18, centos可以忽略,支持centos7.)

安装步骤:

0 首先 按照 tensorflow官方安装教程来

https://tensorflow.google.cn/install/install_linux。

1 安装 gcc 内核

可参考(https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#verify-kernel-packages)

yum -y install gcc kernel-devel kernel-headers

2 添加Repo源

阿里云源 国内更新快
mv /etc/yum.repos.d/CentOS-Base.repo /etc/yum.repos.d/CentOS-Base.repo.backup
wget -O /etc/yum.repos.d/CentOS-Base.repo http://mirrors.aliyun.com/repo/Centos-7.repo

上面源是为了装 nvidia驱动,下面这个是装python相关的。
rpm --import https://www.elrepo.org/RPM-GPG-KEY-elrepo.org
rpm -Uvh http://www.elrepo.org/elrepo-release-7.0-3.el7.elrepo.noarch.rpm
yum install yum-plugin-fastestmirror

python相关源
yum -y install epel-release

yum makecache

3 安装NVIDIA显卡驱动

先安装nvidia-detect,然后运行 nvidia-detect , 会显示显卡对应驱动的版本 kmod-nvidia, 安装检测出来的版本就可以了.

$ yum install nvidia-detect

$ nvidia-detect

$ yum install kmod-nvidia

我按照个过程安装完以后就可以了,有些教程中需要屏蔽 nouveau 弄不好会把系统搞坏。

(也可以手动安装从NVIDIA官网下载驱动安装,期间需要停掉图像界面,中间可能有各种问题,推荐用yum/apt安装)

4 安装CUDA

按照 https://docs.nvidia.com/cuda/cuda-installation-guide-linux/ 官方教程进行即可,安卓cuda最后一步的时候,不要直接 yum install cuda,这样会装到最新版本,需要装。yum install cuda-9.0 然后按照教程添加环境变量到 /etc/profile 和 .bashrc 中

5 安装 cuDNN

下载 https://developer.nvidia.com/cudnn , cudnn 需要注册 nvidia 开发者账号, 然后下载 Download cuDNN v7.0.4 (Nov 13, 2017), for CUDA 9.0, 然后:解压,copy

$ tar -xzvf cudnn-9.0-linux-x64-v7.tgz

$ 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*

 

6 安装 tensorflow

有些可能没有 pip, 先装 python-pip, python-dev,然后pip install tensorflow-gpu 即可,如果中间出现依赖然间包冲突,把低版本的卸载掉重新装。 在pip 安装的时候可以在后面加上 -i http://pypi.mirrors.ustc.edu.cn/simple/ 可以提高速度,以下是几个国内的pip源。

阿里云 http://mirrors.aliyun.com/pypi/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/