Код и tf.keras модели TensorFlow будут прозрачно работать на одном графическом процессоре без необходимости внесения изменений в код. Примечание
xrust: используйте tf.config.list_physical_devices('GPU') для подтверждения того, что TensorFlow использует графический процессор. Самый простой способ запуска на нескольких графических процессорах, на одной или нескольких машинах, — это использование стратегий распределения. Это руководство предназначено для пользователей, которые опробовали эти подходы и обнаружили, что им необходим детальный контроль над тем, как TensorFlow использует ресурсы графического процессора. Чтобы узнать, как устранять проблемы с производительностью в сценариях с одним и несколькими графическими процессорами, см. руководство «Оптимизация производительности графического процессора TensorFlow» . Настраивать Убедитесь, что у вас установлена последняя версия графического процессора TensorFlow. import tensorflow as tf print("Num GPUs Available: ",