2018年8月19日
TensorFlow设置GPU占用量
摘要： tensorflow在训练时默认占用所有GPU的显存。 可以通过以下方式解决该问题： 1、在构造tf.Session()时候通过传递tf.GPUOptions作为可选配置参数的一部分来显式地指定需要分配的显存比例，如下所示： per_process_gpu_memory_fraction指定了每个G 阅读全文
posted @ 2018-08-19 16:36 jiu~ 阅读(11426) 评论(0) 推荐(0) 编辑