В учебнике по Colab по умолчанию используется процессор, поэтому убедитесь, что вы не используете GPU там.Если нет, то посмотрите на мощность вашего ЦП, на colab, ЦП имеет 13 ГБ ОЗУ прибл. Здесь - характеристики колаба.Проблема в основном из-за питания процессора.