安装的tensorflow-gpu与cuda,cudnn,python版本必须相匹配,在下面这个网站查询版本匹配
https://tensorflow.google.cn/install/source#linux
比如我的是python3.6,安装过程:
1.安装cudatoolkit
conda install cudatoolkit=9.0
2.安装cudnn
conda install cudnn=7.0
3.安装tensorflow-gpu
conda install tensorflow-gpu=1.12.0
4.测试是否安装成功,打开python
import tensorflow as tf
print(tf.test.is_gpu_available())
返回True则可用,返回False不可用