spacecruft/nopenpilot
spacecruft/nopenpilot
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

1024 MB should be enough for anyone

Browse source
This commit is contained in:
Willem Melching 2020-09-02 15:12:30 +02:00
parent 9ad9076486
commit 873043b9fb
1 changed files with 1 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
selfdrive/modeld/runners/keras_runner.py
View file

@ -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)