首页 文章

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

提问于
浏览
2

命令

conda install -c anaconda tensorflow-gpu

现在默认安装Tensorflow-gpu 1.7,总是抛出“无法创建会话”的错误

这是截图中的回溯:

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:名称:Tesla M40 24GB major:5 minor: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行,在文件“/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 回答

  • 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) .

相关问题