update docker run file

Terkwood · Mar 23, 2020 · 124b158 · 124b158
1 parent 1c079bd
commit 124b158
Showing 1 changed file with 9 additions and 1 deletion.
diff --git a/tinybrain/run.sh b/tinybrain/run.sh
@@ -1,3 +1,11 @@
 #!/bin/bash
 
-docker run -a stdin -a stdout -a stderr -i --device=/dev/nvhost-ctrl --device=/dev/nvhost-ctrl-gpu --device=/dev/nvhost-prof-gpu --device=/dev/nvmap --device=/dev/nvhost-gpu --device=/dev/nvhost-as-gpu -v /usr/lib/aarch64-linux-gnu:/usr/lib/aarch64-linux-gnu -v /usr/local/cuda-10.0/targets/aarch64-linux/lib:/usr/local/cuda-10.0/targets/aarch64-linux/lib tinybrain
+docker run -a stdin -a stdout -a stderr \
+ -i --device=/dev/nvhost-ctrl \
+ --device=/dev/nvhost-ctrl-gpu \
+ --device=/dev/nvhost-prof-gpu \
+ --device=/dev/nvmap --device=/dev/nvhost-gpu \
+ --device=/dev/nvhost-as-gpu \
+ -v /usr/lib/aarch64-linux-gnu:/usr/lib/aarch64-linux-gnu -v /usr/local/cuda-10.0/targets/aarch64-linux/lib:/usr/local/cuda-10.0/targets/aarch64-linux/lib \
+ -v ${PWD}/.env:/var/BUGOUT/.env \
+ tinybrain