墨塵的瑣碎知識點記錄--9: CUDA_ERROR_OUT_OF_MEMORY

報錯信息:

在這裏插入圖片描述

1.代碼頭部添加:

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
2.限制當前腳本可用顯存
gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章