tensorflow安装（GPU版本）

安装前准备

centos7安装英伟达显卡驱动的方法
cuda安装
cudnn安装

tensorflow版本选择

目前我们安装的英伟达驱动是418.43
cuda:10.1
cudnn:7.5
如果安装tensorflow_gpu==1.60的话，会安装成功，
在python引入tensorflow的时候报错
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
意思为tensorflow_gpu依赖于cuda9.0，但是我们安装的是10.1
解决办法有两个
一：升高tensorflow版本，比如我安装的是最新的1.13
二：降低cuda的版本，当然driver和cudnn也需要相应改变（**不建议**）

参考：https://pypi.org/project/tensorflow-gpu/1.6.0/#files

在线安装

pip install --upgrade tensorflow-gpu

本地安装

pip install ./tensorflow_gpu-1.6.0-cp36-cp36m-manylinux1_x86_64.whl 

验证是否安装成功

>>> tf.__version__
'1.13.1'
import tensorflow as tf
with tf.device('/gpu:1'):  
    a = tf.constant([1.0, 2.0, 3.0, 4.0, 5.0], shape=[3, 4], name='a')  
    b = tf.constant([2.0, 4.0, 6.0, 8.0, 10.0], shape=[4, 3], name='b')  
    c = tf.matmul(a, b) 
sess=tf.Session(config=tf.ConfigProto(log_device_placement=True))
print(sess.run(c))

卸载

pip uninstall tensorflow_gpu