原文: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))