Что касается памяти графического процессора, см. Этот вопрос (решение подпроцесса и сброс памяти графического процессора numba у меня работали раньше):
Обычно память CPU используется для данных процессора GPU. передача, так что тут ничего не поделаешь, но вы можете иметь больше памяти с помощью простого трюка:
a=[]
while True:
a.append('qwertyqwerty')
время выполнения colab остановится и даст вам возможность увеличить память.
happy deep обучения!