sess.run 問題
2020/01/14 06:30 下午
電腦視覺深度學習討論版
胡連福
觀看數:1
回答數:3
收藏數:0
cvdl-1
cvdl-1-d35
請問老師 sess.run 問題 :
1.我分別在 colab 與 我的電腦上跑一樣的code, 當執行到底下這段 sess.run 時, 我分別用 nvidia-smi 觀察 GPU記憶體, 為何 GPU 所佔的記憶體大小不一樣 ?
2.我的電腦是已啟用 CUDA, GPU記憶體有 2G, 我的電腦GPU記憶體應該不夠, 但為何沒提示GPU記憶體不夠的 error, 而可以繼續跑完 code ?
(1) 在 colab 用 nvidia-smi 觀察 GPU記憶體使用了8G, 如下 :
(2) 在 我的電腦用 nvidia-smi 觀察 GPU記憶體只佔了64M (感覺好像沒動作), 如下 :
(3) code 如下 :
(3.1)
#輸入網絡架構的參數檔
saver.restore(sess, 'models/pretrain/yolo_tiny.ckpt')
#開啟對話,進行預測
np_predict = sess.run(predicts, feed_dict={image: np_img})
(3.2)
print("預測結果和標記框的損失量")
with tf.Session() as sess1:
print(sess1.run(output_loss))