遇到一个问题:
新买显卡显存8G但是tensorflow运行的时候界面显示只有约6.3G的显存可用,如下图:
即限制了我的显存,具体原因为什么我也不知道,但原来的显存小一些的显卡就没有这个问题。
目前的解决办法是:
官方文档解决
然后对应的中文博客
总结一下,就是下面的两个办法:
设置每次使用时按需申请
# 获取所有GPU组成list
gpus = tf.config.experimental.list_physical_devices('GPU')
# 设置按需申请
# 由于我这里仅有一块GPU,multi-GPU需要for一下
tf.config.experimental.set_memory_growth(gpu[0], True)
直接限制可使用显存大小
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)])
最后,这两个方法的不足,即还没有完全解决我的问题:
但是这两个办法都是只能修改本次运行,
如何设置为全局设置,即每次打开都不用再添加这几行代码就为上次的设置,
我目前还没发现。。。
如有看到的请教我一下嘛