【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

>>阅读原文<<