from tensorflow.keras.models import load_model model = load_model(r"D:\K_REPO\ComV\AI_made\trainnig_output\final_model_2.h5") model.summary() from tensorflow.python.client import device_lib print("[INFO] Devices available:") print(device_lib.list_local_devices())