tensorflow1中如何设置GPU的使用
生活随笔
收集整理的這篇文章主要介紹了
tensorflow1中如何设置GPU的使用
小編覺得挺不錯的,現(xiàn)在分享給大家,幫大家做個參考.
import tensorflow as tf #tf.ConfigProto()主要的作用是配置tf.Session的運算方式
config = tf.ConfigProto()
config.gpu_options.allow_growth = True # 不全部占滿顯存, 按需分配
config.gpu_options.per_process_gpu_memory_fraction = 0.3 # 限制GPU內(nèi)存占用率
sess = tf.Session(config=config)
總結(jié)
以上是生活随笔為你收集整理的tensorflow1中如何设置GPU的使用的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。
- 上一篇： 如何在tensorflow2环境运行te
- 下一篇： tensorflow2中如何设置GPU的