【Ubuntu-Tensorflow】GPU设置及显存资源分配

最近笔者在做GPU显存资源分配的研究，发现在tf中gpu一些实用的方法和接口，共享出来，供大家参考学习，如有问题，欢迎留言讨论。 1.运行程序时，控制台设置GPU运行参数，占用显存全部资源 $ CUDA_VISIBLE_DEVICES=0 python predict.py # 只使用gpu：0设备运行predict.py程序。 $ CUDA_VISIBLE_DEVICES=1 python pr

>>阅读原文<<