1024 MB should be enough for anyone
This commit is contained in:
parent
9ad9076486
commit
873043b9fb
|
@ -41,10 +41,9 @@ if __name__ == "__main__":
|
|||
# limit gram alloc
|
||||
gpus = tf.config.experimental.list_physical_devices('GPU')
|
||||
if len(gpus) > 0:
|
||||
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=2048)])
|
||||
tf.config.experimental.set_virtual_device_configuration(gpus[0], [tf.config.experimental.VirtualDeviceConfiguration(memory_limit=1024)])
|
||||
|
||||
m = load_model(sys.argv[1])
|
||||
print(m, file=sys.stderr)
|
||||
|
||||
run_loop(m)
|
||||
|
||||
|
|
Loading…
Reference in a new issue