[問題] 同時執行多個tensorflow程式消失

看板Python作者時間6年前 (2017/09/06 00:22), 6年前編輯推噓3(306)
留言9則, 3人參與, 最新討論串1/1
大家好,我現在用tensorflow訓練好兩個模型,分別用在兩個程式裡,平常單獨用的時候 都ok,但是我如果同時使用兩個程式就會變成一個正常一個報錯,請問是因為tensorflow 只能使用一個嗎? 這個是執行的時候顯示的 https://i.imgur.com/hIwsdGF.jpg
-- ※ 發信站: 批踢踢實業坊(ptt.cc), 來自: 101.9.178.164 ※ 文章網址: https://www.ptt.cc/bbs/Python/M.1504628530.A.4E2.html

09/06 00:29, , 1F
OOM(Out of memory) 記憶體不足?
09/06 00:29, 1F

09/06 02:27, , 2F
nvidia-smi看一下gpu記憶體用量
09/06 02:27, 2F
我看了,10761/11169所以代表我這台電腦注定只能用一個了? ※ 編輯: aa12twtwaa (101.9.178.164), 09/06/2017 11:09:22

09/06 16:54, , 3F
關鍵字tensorflow gpu memory management
09/06 16:54, 3F

09/06 16:54, , 4F
調整程式gpu memory的用量
09/06 16:54, 4F

09/06 16:56, , 5F
gpu_options = tf.GPUOptions(per_process_gpu_memor
09/06 16:56, 5F

09/06 16:56, , 6F
y_fraction=0.45) 之類的
09/06 16:56, 6F

09/06 16:56, , 7F
sess = tf.Session(config=tf.ConfigProto(gpu_optio
09/06 16:56, 7F

09/06 16:56, , 8F
ns=gpu_options))
09/06 16:56, 8F

09/06 20:00, , 9F
成功了,感謝樓上!?
09/06 20:00, 9F
文章代碼(AID): #1PhiyoJY (Python)