Tensorflow-gpu(1.7)Conda安装 - 无法创建会话

时间:2018-04-26 11:12:22

标签: session tensorflow cuda keras

命令

conda install -c anaconda tensorflow-gpu 

现在默认安装Tensorflow-gpu 1.7,总是抛出错误#34;无法创建会话"

以下是屏幕截图中的回溯:

import tensorflow as tf
tf.Session()
  

2018-04-26 16:34:37.971115:I tensorflow / core / platform / cpu_feature_guard.cc:140]您的CPU支持未编译此TensorFlow二进制文件的指令:SSE4.1 SSE4.2 AVX AVX2 FMA   2018-04-26 16:34:38.588037:I tensorflow / core / common_runtime / gpu / gpu_device.cc:1344]找到具有属性的设备0:   名称:特斯拉M40 24GB专业:5小调:2 memoryClockRate(GHz):1.112   pciBusID:0000:02:00.0   totalMemory:22.40GiB freeMemory:10.62GiB   2018-04-26 16:34:38.588119:I tensorflow / core / common_runtime / gpu / gpu_device.cc:1423]添加可见的gpu设备:0   2018-04-26 16:34:38.604343:E tensorflow / core / common_runtime / direct_session.cc:167]内部:cudaGetDevice()失败。状态:CUDA驱动程序版本不足以用于CUDA运行时版本   Traceback(最近一次调用最后一次):     文件"",第1行,in     文件" /home/16mcmi17/anaconda3/envs/myEnv/lib/python3.5/site-packages/tensorflow/python/client/session.py" ;,第1509行, init       super(Session,self)。 init (target,graph,config = config)     文件" /home/16mcmi17/anaconda3/envs/myEnv/lib/python3.5/site-packages/tensorflow/python/client/session.py" ;,第638行, init       self._session = tf_session.TF_NewDeprecatedSession(opts,status)     文件" /home/16mcmi17/anaconda3/envs/myEnv/lib/python3.5/site-packages/tensorflow/python/framework/errors_impl.py" ;,第516行,退出       c_api.TF_GetCode(self.status.status))   tensorflow.python.framework.errors_impl.InternalError:无法创建会话。

早期的tensorflow-gpu 1.4工作正常。

我想知道问题是与版本有关还是其他? 另外,如何使用特定版本的conda(非pip)安装tensorflow-gpu?

1 个答案:

答案 0 :(得分:2)

根据你追踪的这一部分:

  

内部:cudaGetDevice()失败。状态:CUDA驱动程序版本是   不适合CUDA运行时版本

您似乎无法满足CUDA版本要求。 Tensorflow 1.4需要CUDA 8.0+,而Tensorflow 1.7需要CUDA 9.0 +。

尝试将您的CUDA升级到9.0(CUDA installation guide for Linux)。