阿里云gpu服务器安装TensorFlow-gpu过程记录

阿里云gpu服务器：

为了少踩坑，这里默认选择了自动安装GPU驱动，版本选9.0.176下的390.46；

ps：这里是在下单选操作系统的时候选择，后来打算重装系统的时候发现重装没有选驱动这一步，遂作罢未重装，不知道重装提交工单给不给装；系统开通后进去就能看到装驱动的log文件，成功即可；

系统更新：

yum update

https://developer.nvidia.com/cudnn官网下载cuDNN，需要注册账号登陆，下载版本 Download cuDNN v7.3.1 (Sept 28, 2018), for CUDA 9.0下的cuDNN v7.3.1 Library for Linux；

外国网站国内下载可能较慢，着急可用国外vps中转；

tar -zxvf cudnn-9.0-linux-x64-v7.3.1.20.tgz
cd cuda
cp include/* /usr/local/cuda-9.0/inlcude
cp lib64/lib* /usr/local/cuda-9.0/lib64

修改环境变量，执行

sudo vim /etc/profile

在export PATH 那行下面加上两行代码：

export PATH=/usr/local/cuda-9.0/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH

退出执行

source /etc/profile

pip3 install tensorflow-gpu==1.12.0

执行测试代码，看是否能够正常导入，并输出自定义的文字：

import tensorflow as tf

hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
sess.run(hello)

之后执行

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())

检测到gpu即成功。

参考编辑：https://blog.csdn.net/Oh_My_Fish/article/details/78861867