sess.run 問題 - Cupoy

2020/01/14 10:30 AM

電腦視覺深度學習論壇

胡連福

觀看數：0

回答數：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))