У меня тоже была такая же проблема при использовании Google Colab. Поэтому я импортировал все модули из модулей tenorflow.keras вместо традиционного метода вызова только через keras.
Например. вместо from keras.models import Sequential
использование
from tensorflow.keras import Sequential
А, from keras.layers import Conv2D, MaxPool2D, Dense, Flatten, Dropout
использование
from tensorflow.keras.layers import Conv2D, MaxPool2D, Flatten, Dropout, Dense
Итак, мой новый код выглядит следующим образом. Мой новый код.