我驗證了我的 TF 正確使用了我的 GPU。TF 使用 GPU 來運行 model.fit(),在 HWiNFO64 中看到大約 50% 的使用率。但是,如果我隨后將此單元添加到筆記本中,筆記本將使用該模型來預測測試集中圖像的標簽: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 和其他進程使用)。有沒有辦法讓我在 GPU 上運行 model.predict(),或者是否有任何替代方案可以為單個輸入提供模型輸出?
如何使用GPU運行Keras Model.Predict()
寶慕林4294392
2024-01-04 17:11:30