rgal

Which script are you using? train_keras_network.py? I haven’t tried this on GPU. Can you see where different operations are running with logging device placement (https://www.tensorflow.org/guide/gpu#logging_device_placement)?