1.核对显卡CUDA,CUDNN版本,选择相应版本tensorflow
参考https://blog.csdn.net/qq_27825451/article/details/89082978
2.添加环境变量(否则报错ImportError: libcublas.so.9.0: cannot open shared object file: No such file or director)
参考https://blog.csdn.net/ksws0292756/article/details/80034086
3 安装
pip3 install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple
或者
pip install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple