tensorflow2.0显存设置

遇到一个问题:
新买显卡显存8G但是tensorflow运行的时候界面显示只有约6.3G的显存可用,如下图:
在这里插入图片描述
即限制了我的显存,具体原因为什么我也不知道,但原来的显存小一些的显卡就没有这个问题。

目前的解决办法是:
官方文档解决
然后对应的中文博客

总结一下,就是下面的两个办法:

  1. 设置每次使用时按需申请

    # 获取所有GPU组成list
    gpus = tf.config.experimental.list_physical_devices('GPU')
    # 设置按需申请
    # 由于我这里仅有一块GPU,multi-GPU需要for一下
    tf.config.experimental.set_memory_growth(gpu[0], True)
    
  2. 直接限制可使用显存大小

    gpus = tf.config.experimental.list_physical_devices('GPU')
    # 对需要进行限制的GPU进行设置
    tf.config.experimental.set_virtual_device_configuration(gpus[0],
                                                            [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=8192)])
    

最后,这两个方法的不足,即还没有完全解决我的问题:
但是这两个办法都是只能修改本次运行,
如何设置为全局设置,即每次打开都不用再添加这几行代码就为上次的设置,
我目前还没发现。。。
如有看到的请教我一下嘛

你可能感兴趣的:(python,tf2.0)