The kernel for MyTest.ipynb appears to have died. It will restart automatically.

[HBarnabas]

Hello,

I'm trying to run a test jupyter notebook of a LSTM model running tensorflow. I have tried setting the GPU memory limit like suggested here. But still the I get the error mentioned above. I can not find anything realted to GC vertex AI and everyone suggest setting the gpu memory in case of such errors. 

For reference I have tried to run this as well on my Vertex AI jupyter lab and it crashes as well. The only thing I added was this:

gpus = tf.config.list_physical_devices('GPU')
if gpus:
tf.config.set_logical_device_configuration(
gpus[0],
[tf.config.LogicalDeviceConfiguration(memory_limit=12288)]
)

logical_gpus = tf.config.list_logical_devices('GPU')
print(len(gpus), "Physical GPU,", len(logical_gpus), "Logical GPUs")

On my personal computer it runs just fine, but it would take 13 hours to train which is not a option for me at the moment.

Any help would be appriciated. 

Barnabas.