当前tensorflow官方版本 1.7 支持CUDA9.0. 并未支持CUDA 9.1。
如果需要自己编译,参考:https://blog.csdn.net/u013832707/article/details/73161071
编译环境:
ubuntu 16.04, python 2.7, cuda9.1, cudnn v7.1
tensorflow-1.7.0rc1-cp27-cp27mu-linux_x86_64-xin-cuda9.1-cudnn7.1.whl
链接:https://pan.baidu.com/s/1t9v0vwP8S5CW9d6wdL43Fw 密码:b8wj
sudo pip install tensorflow-1.7.0rc1-cp27-cp27mu-linux_x86_64-xin-cuda9.1-cudnn7.1.whl