tf.config.experimental.list_logical_devices('GPU')使内核和Jupyter笔记本服务器崩溃

时间:2019-10-02 19:34:45

标签: python tensorflow keras jupyter-notebook tf.keras

我面临一个奇怪的问题。

我的tensorflow版本是1.14.0,tensorflow-gpu版本是1.14.0 我的keras版本是2.3.0。

我有5 gpus。

enter image description here Python版本是3.6

我的NVCC版本是10.0

我的cuDNN版本是7.6

#define CUDNN_MAJOR 7
#define CUDNN_MINOR 6
#define CUDNN_PATCHLEVEL 1
--
#define CUDNN_VERSION (CUDNN_MAJOR * 1000 + CUDNN_MINOR * 100 + CUDNN_PATCHLEVEL)

#include "driver_types.h"

问题在于

logical_gpus=tf.config.experimental.list_logical_devices('GPU')

jupyter笔记本中的此单元使整个jupyter服务器崩溃。

到目前为止,我还没有提出任何解决方案。

请帮助。

0 个答案:

没有答案