有没有什么方法可以在许多jupyter笔记本中同时在GPU上运行keras with tensorflow后端?

时间:2018-03-08 16:12:22

标签: tensorflow cuda keras jupyter-notebook jupyter

场景是:

  1. 我在notebookA中运行keras。
  2. 我在notebookB中运行keras。
  3. 当我开始在notebookB中训练我的模型时,我得到InternalError: Blas GEMV launch failed
  4. 我知道这里有类似的问题TensorFlow: InternalError: Blas SGEMM launch failed我可以通过关闭另一个使用gpu的笔记本(notebookA)来解决这个问题。

    但是我想知道是否有同时在多个笔记本中运行带有tensorflow后端的keras?或者现在暂时无法实现这一目标?

    我尝试在每个笔记本中使用下面的设置,但它没有工作(得到了同样的错误)。

    import tensorflow as tf
    from keras.backend.tensorflow_backend import set_session
    config = tf.ConfigProto()
    config.gpu_options.per_process_gpu_memory_fraction = 0.3
    set_session(tf.Session(config=config))
    

    非常感谢!

0 个答案:

没有答案