tf.ConfigProto (allow_soft_placement=True)

tf.ConfigProto一般用在创建session的时候用来对session进行参数配置。
with tf.Session(config = tf.ConfigProto(....)....)
tf.ConfigProto()的参数

   log_device_placement = True : 是否打印设备分配日志
   allow_soft_placement = True : 如果你指定的设备不存在,允许TF自动分配设备
   tf.ConfigProto(log_device_placement = True, allow_soft_placement = True)

控制GPU资源使用率

allow growth

  config = tf.ConfigProto()
  config.gpu_options.allow_growth = True
  session = tf.Session(config = config,....)

使用allow_growth option,刚一开始分配少量的GPU容量,然后按需慢慢的增加,由于不会释放内存,所以会导致碎片
per_process_gpu_memory_fraction

   gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction = 0.7)
   config = tf.ConfigProto(gpu_options = gpu_options)
   session = tf.Session(config = config,....)

设置每个GPU应该拿出多少容量给进程使用,0.4代表40%


控制使用哪块GPU

 ~/CUDA_VISBLE_DEVICES = 0      python your.py #使用GPU0
 ~/CUDA_VISBLE_DEVICES = 0,1   python your.py #使用GPU0,1

注意单词不要打错

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 1、tf.shape(varA)和varA.get_shape()这两个API都是返回varA的size大小,但是...
    溪奇阅读 9,189评论 0 1
  • 1. 使用指定的GPU和显存 如果设备上装备了多块GPU,TF运行时默认使用所有与他可见GPU,而且默认使用尽可能...
    肆不肆傻阅读 5,337评论 0 0
  • 这篇文章是针对有tensorflow基础但是记不住复杂变量函数的读者,文章列举了从输入变量到前向传播,反向优化,数...
    horsetif阅读 4,890评论 0 1
  • 花开无语,芳华烁烁;花落无言,余香阵阵。虽然做不到看破红尘无悲无喜得失两忘,也要修炼一颗淡定从容的心,泰然面对日出...
    拾光缩写阅读 3,093评论 0 0
  • 自从《反贪风暴2》上映之后,港片的热潮又一次引发了媒体和影迷的热议更甚有说:每位主演研制和演技爆棚,汇聚的是那些年...
    简娯阅读 2,999评论 0 2