inception_net在GPU环境下jupyter notebook会挂掉，在CPU环境下正常。

来源：4-6 Inception-mobile_net（1）

despacito_458

2019-11-09

请问老师，这里是什么原因啊？

如果是环境的差异，在GPU环境下怎么调整呢？

写回答

1回答

正十七

2019-11-17

一般情况下，这种错误是因为GPU memory不够用了，因为默认一个进程会占据全部的GPU显存，如果重启，可能就会有不够用的情况。你可以监控下GPU的显存是不是用光了。

如果确实是这个问题，需要在tensorflow代码中添加这样的代码：

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
sess = tf.Session(config=config)

这个代码的作用就是每个进程需要多少显存就用多少。这样可以让GPU坚持的久一些。

深度学习之神经网络(CNN/RNN/GAN)算法原理+实战

深度学习算法工程师必学，深入理解深度学习核心算法CNN RNN GAN

2617 学习 · 935 问题

相似问题

回答 1

回答 1

回答 1

回答 1

回答 1