Add Dockerfile.nvidia

2023-06-16 09:14:14 -04:00
parent 8f42e11e43
commit 47cc73313b
2 changed files with 26 additions and 0 deletions
--- a/docker/run.sh
+++ b/docker/run.sh
@@ -0,0 +1,6 @@
+#!/bin/bash
+# Run this script from within the refacer/docker folder.
+
+docker build -t refacer -f Dockerfile.nvidia . && \
+    docker run --rm -p 7860:7860 --gpus all refacer python3 app.py --server_name 0.0.0.0 && \
+    sleep 1 && google-chrome --new-window "http://127.0.0.1:7860" &