在ubuntu16.04上配置tensorflow-gpu
所配环境：ubuntu16.04+Anaconda3+python3.5.4+CUDA10.1+cudnn7.6.5
anaconda3、pycharm的安装不在本文赘述。参考文章：https://blog.csdn.net/yiyayiya557/article/details/105641706
我在anaconda3的虚拟环境进行配置，便于我管理不同的深度学习框架，例如tensorflow、pytorch、theano。
1. 建立tensorflow_gpu环境
安装好anaconda3后，创建tensorflow虚拟环境，python版本必须为3.5或3.6，不然keras安装会报错，同时为了与windows的开发环境保持一致，选择python 3.5.4
conda create -n tensorflow python=3.5.4建议配置为镜像环境，例如清华镜像，pip安装会快很多，不然很慢。
查看所有环境
conda env list
conda info --envs删除环境
conda env remove -n tensorflow激活环境
conda activate tensorflow进入tensorflow虚拟环境，先安装tensorflow-cpu
pip install tensorflow确认tensorflow已经安装成功，不报错，就说明tensorflow-cpu已经安装成功。
python
import tensorflow2. CUDA下载及安装
到这里，基本都已经安装好显卡驱动了，如果有问题，参考：https://blog.csdn.net/yiyayiya557/article/details/105574008
（1）确认是否安装gcc
gcc --version（2）确认nouveau已禁用，如果没有输出则说明已禁用。
lsmod | grep nouveau输入nvidia-smi，有如下显示则说明驱动安装成功：
然后确认CUDA与cudnn配对版本，并下载适合自己系统的版本
https://blog.csdn.net/omodao1/article/details/83241074
我下载了cuda10.1， cudnn7.6.5
下载地址：https://developer.nvidia.com/cuda-downloads?target_os=Linux
建议下载run文件，安装比较方便。
输入
wget http://developer.download.nvidia.com/compute/cuda/10.1/Prod/local_installers/cuda_10.1.243_418.87.00_linux.run
sudo sh cuda_10.1.243_418.87.00_linux.run选择accept EULA，不安装显卡驱动，其他用默认设置，如下
配置系统环境
sudo gedit ~/.bashrc在文件末尾输入
export PATH="/usr/local/cuda-10.1/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-10.1/lib64:$LD_LIBRARY_PATH"
export CUDA_HOME="/usr/local/cuda-10.1"
重启环境
source ~/.bashrc
nvcc --version如果显示CUDA版本信息代表正常。
测试sample例子：
#编译并测试设备 deviceQuery：
cd ~/NVIDIA_CUDA-10.1_Samples/1_Utilities/deviceQuery
make deviceQuery
./deviceQuery
#编译并测试带宽 bandwidthTest：
cd ../bandwidthTest
make bandwidthTest
./bandwidthTest如果这两个测试的最后结果都是Result = PASS，说明CUDA安装成功。
3. cudnn下载及配置
下载地址：https://developer.nvidia.com/rdp/cudnn-archive
下载cudnn-10.1-linux-x64-v7.6.5.32.tgz，提取到本地。
将库(.lib)和头文件(.h)copy到cuda目录
sudo cp /home/robert/下载/cuda/include/cudnn.h /usr/local/cuda-10.1/include
sudo cp /home/robert/下载/cuda/lib64/libcudnn* /usr/local/cuda-10.1/lib64修改文件访问权限：
sudo chmod a+r /usr/local/cuda-10.1/include/cudnn.h
sudo chmod a+r /usr/local/cuda-10.1/lib64/libcudnn*到这里tensorflow配置完成。
测试tensorflow，输入conda activate tensorflow，
库文件打不开，解决办法参考文章：https://blog.csdn.net/yiyayiya557/article/details/105700696
如代码所示：
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
参考文章：https://blog.csdn.net/qq_38901147/article/details/90049666