Добавить файл зависимости tensorflow-gpu==2.0 в setup.py для обучения с Tensorflow 2.0 в версии 1.15 для платформы AI Platform. Tensorflow 2.1 не может использовать графические процессоры из-за несоответствия в библиотеках CUDA, но вы можете тренироваться с процессорами. Будьте осторожны, чтобы другие зависимости не перезаписывали tensorflow-gpu. Отказ от ответственности: я обучил только одну работу таким образом, и это действительно хакерское решение.