Viewing a single comment thread. View all comments

veb101 t1_j0atf1m wrote

Reply to comment by MrFlufypants in [D] Tensorflow vs. PyTorch Memory Usage by Oceanboi

I think this can be solved using:

tf.config.experimental.set_memory_growth(gpu_device, True)

11

MrFlufypants t1_j0atiwu wrote

There are a couple ways to do it. That’s the one I use normally. Sometimes that doesn’t work though. Can’t quite remember the use case where it wasn’t working

1

MOSFETBJT t1_j0dbib4 wrote

This is what helped when I had a similar issue

1