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

针对新购显卡在TensorFlow中显存显示不全的问题,本文介绍了两种解决方案:按需分配显存和直接限制显存大小,并探讨了如何将设置全局化。
471

被折叠的 条评论
为什么被折叠?



