Sau khi bạn cài đặt phiên bản GPU của Tensorflow, bạn không có bất cứ điều gì để làm trong Keras. Như được viết trong tài liệu Keras , "Nếu bạn đang chạy trên phụ trợ TensorFlow, mã của bạn sẽ tự động chạy trên GPU nếu phát hiện thấy bất kỳ GPU có sẵn nào."
Và nếu bạn muốn kiểm tra xem GPU có được phát hiện chính xác hay không, hãy bắt đầu tập lệnh của bạn với:
import tensorflow as tf
sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))
Đầu ra tiêu chuẩn không được hiển thị bất kỳ lỗi nào và in tên của GPU. Nếu vậy, bạn đã sẵn sàng để chạy Keras và Tensorflow ở chế độ GPU.