墨尘的琐碎知识点记录--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))
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章