学習時にGPUを使って学習はできたのですが、予測時にGPUを使うことができていません。
以下のコードをCPUで実行しています。
Python
1model = load_model('model path') 2files = sorted(os.listdir('image directory path')) 3for i,f in enumerate(files): 4 img = load_img('directory path'+f, target_size=(224,224)) 5 array = img_to_array(img) 6 x = np.expand_dims(array, axis=0) 7 x = preprocess_input(x) 8 x = np.rollaxis(x, 1, 4) 9 ret = model.predict(x)
そもそも予測時にGPUを用いることはできるのでしょうか?
回答1件
あなたの回答
tips
プレビュー
バッドをするには、ログインかつ
こちらの条件を満たす必要があります。
2018/07/19 00:52