SSD讲堂五(训练)_错误调试：failed to allocate 4.75G (5102921216 bytes) from device: CUDA_ERROR_OUT_OF_MEMORY

原文：https://blog.csdn.net/strive_for_future/article/details/81292865?utm_source=copy

1）需要指定GPU，代码头部添加如下代码：

import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"

2）限制当前脚本可用显存，代码头部添加第一行，session 语句进行如第二行的修改

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

 
 

你好博主，这个方法确实有效果

1）我方在了代码头部

2）里面的第一句话我用上了替换了原来的对应代码

请问最后的这句我怎么在我的程序里面没有看到

sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

下面是我的情况

全