tensorflow gpu在LINUX上运行失败

我已经在ubuntu 16.04上安装了CUDA和cuDnn

CUDA版本：9.0//带有驱动程序版本390.87

cuDNN版本：适用于CUDA9.0的7.2

import tensorflow as tf

很好，但是

tf.Session() 

呈现以下错误

2018-09-15 16:43:23.281375: I tensorflow/core/common_runtime/gpu/gpu_device.cc:1484] Adding visible gpu devices: 0
2018-09-15 16:43:23.281431: E tensorflow/core/common_runtime/direct_session.cc:158] Internal: cudaGetDevice() failed. Status: CUDA driver version is insufficient for CUDA runtime version
Traceback (most recent call last):
 File "<stdin>", line 1, in <module>
 File "/home/imhgchoi/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 1494, in __init__
super(Session, self).__init__(target, graph, config=config)
 File "/home/imhgchoi/anaconda3/lib/python3.6/site-packages/tensorflow/python/client/session.py", line 626, in __init__
self._session = tf_session.TF_NewSession(self._graph._c_graph, opts)
tensorflow.python.framework.errors_impl.InternalError: Failed to create session.

---

也许是你的环境变量导致了这个问题。 试试这个：

在~/.bashrc文件的末尾添加这些行，打开一个终端，在那里启动一个python会话，然后导入tensorflow（您应该通过apt安装tensporflow gpu），看看它是否工作：

sudo vim ~/.bashrc

并在文件末尾添加以下内容，然后重新启动终端：

export CUDA_HOME="/usr/local/cuda-9.0"
export LD_LIBRARY_PATH="${CUDA_HOME}/lib64"
export PATH="${CUDA_HOME}/bin:${PATH}"
export DYLD_LIBRARY_PATH="${CUDA_HOME}/lib"

编辑：1

请确保“usr/local/cuda-9.0”是您安装cuda的目录。

添加了错误..：（我已经更新了帖子上的错误。nvcc——版本在你的终端中返回了什么？@hyongyufroilanchoi）

export CUDA_HOME="/usr/local/cuda-9.0"
export LD_LIBRARY_PATH="${CUDA_HOME}/lib64"
export PATH="${CUDA_HOME}/bin:${PATH}"
export DYLD_LIBRARY_PATH="${CUDA_HOME}/lib"