我驗證了我的 TF 正確使用了我的 GPU。TF 使用 GPU 來運行 model.fit()，在 HWiNFO64 中看到大約 50% 的使用率。但是，如果我隨后將此單元添加到筆記本中，筆記本將使用該模型來預(yù)測測試集中圖像的標(biāo)簽：import numpy as npfor img in test_images:? ? prediction = model.predict(np.expand_dims(img, axis=0)) # Here? ? print(class_names[np.argmax(prediction)])我看到 GPU 使用率只有 1%（Chrome 和其他進(jìn)程使用）。有沒有辦法讓我在 GPU 上運行 model.predict()，或者是否有任何替代方案可以為單個輸入提供模型輸出？