Ubuntu18.04+CUDA9.1+cuDNN7.0.5下安装tensorflow-gpu版本

由于我的NVIDIA 驱动是390:

Ubuntu18.04+CUDA9.1+cuDNN7.0.5下安装tensorflow-gpu版本

所以我开始以为安装CUDA9.0(一直包CUDA和驱动版本不匹配),其实根据下表应该安装CUDA9.1。

Ubuntu18.04+CUDA9.1+cuDNN7.0.5下安装tensorflow-gpu版本

最后我安装好了CUDA9.1+cuDNN7.0.5后安装tensorflow-gpu死活装不上,我试过tensorflow-gpu的版本有:1.5.0, 1.6.0 ,

1.7.0, 1.8.0, 1.10.0, 1.12.0, ...最后用1.6.0终于成功,import tensorflow as tf不报错, 运行sess = tf.Session()也不报错。

这就证明安装好了。