版本
tensorflow-gpu 版本 1.11.0
CUDA 版本 9.0(注:pip安装的tensorflow只支持9.0版本,不支持最新10.0以及9.1、9.2)
cudnn 7.3 (for CUDA 9.0)
下载链接
CUDA 9.2 https://developer.nvidia.com/cuda-92-download-archive?target_os=Linux
CUDA 9.0 https://developer.nvidia.com/cuda-90-download-archive?target_os=Linux
cudnn 7.3 https://developer.nvidia.com/rdp/cudnn-download (要注册开发者帐号,啰嗦无比)
屏蔽默认带有的nouveau(默认 GPU 驱动)
1、打开/lib/modprobe.d/dist-blacklist.conf
2、编辑该文件,操作如下:
将nvidiafb注释掉: #blacklist nvidiafb
然后添加以下语句:
blacklist nouveau
options nouveau modeset=0
3、重建initramfs image步骤
mv /boot/initramfs-(uname -r).img.bak
dracut /boot/initramfs-(uname -r)
4、修改运行级别为文本模式
systemctl set-default multi-user.target
5、重新启动, 使用root用户登陆
reboot
6、查看nouveau是否已经禁用
lsmod | grep nouveau
准备工作
建议参考官方文档 https://developer.download.nvidia.com/compute/cuda/9.2/Prod2/docs/sidebar/CUDA_Installation_Guide_Linux.pdf
安装 dkms
yum install epel-release
yum install --enablerepo=epel dkms
安装 kernel 的相关包
yum install kernel*
下载 CUDA 的直接可执行包
下载 CUDA 9.0 和 CUDA 9.2 (这里不下载 rpm 包)
安装和运行 CUDA 9.2 安装 NVIDIA Driver 和 toolkit
sh cuda_9.2.148_396.37_linux
验证 Nvidia 驱动
dkms status
nvidia-smi
安装CUDA 9.0
1、卸载 CUDA 9.2 的 toolkit
/usr/local/cuda-9.2/bin/uninstall_cuda_9.2.pl
2、安装 CUDA 9.0 但是不要安装 NVIDIA Driver (执行中有选项)
sh cuda_9.0.176_384.81_linux-run
验证驱动
./usr/local/cuda-9.1/extras/demo_suite/deviceQuery
安装 cudnn 7.3 参考官方文档
https://docs.nvidia.com/deeplearning/sdk/cudnn-install/index.html
一共4步
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*
设置路径
添加 /usr/local/cuda-9.0/bin 到 PATH
添加 /usr/local/cuda-9.0/lib64 到 LD_LIBRARY_PATH
具体参考官方文档里面的命名建议 https://developer.download.nvidia.com/compute/cuda/9.2/Prod2/docs/sidebar/CUDA_Installation_Guide_Linux.pdf
安装 tensorflow
pip install tensorflow-gpu (默认安装1.11.0)
代码验证
import tensorflow as tf
sess = tf.Session()
常见问题
1、执行 nvidia-smi,一直报 “NVIDIA-SMI has failed because it couldn't communicate with the NVIDIA driver ...... ”
NVIDIA 驱动安装不正确,需要重装,先执行 nvidia-uninstall 卸载,再重装驱动
2、libcublas.so.9.0: cannot open shared object file: No such file or directory
LD_LIBRARY_PATH 未正确设置,或者安装了9.2的CUDA(因此没找到9.0的 so 文件)
其他
yum list
yum install