您好,欢迎光临本网站![请登录][注册会员]  

搜索资源列表

  1. tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现

  2. 今天小编就为大家分享一篇tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧
  3. 所属分类:其它

    • 发布日期:2020-09-18
    • 文件大小:26624
    • 提供者:weixin_38663452
  1. tensorflow:指定gpu 限制使用量百分比,设置最小使用量的实现

  2. 在Python代码中指定GPU import os os.environ[CUDA_VISIBLE_DEVICES] = 0 设置定量的GPU使用量: config = tf.ConfigProto() config.gpu_options.per_process_gpu_memory_fraction = 0.9 # 占用GPU90%的显存 session = tf.Session(config=config) 设置最小的GPU使用量: config = tf.ConfigProto
  3. 所属分类:其它

    • 发布日期:2021-01-21
    • 文件大小:33792
    • 提供者:weixin_38639615