Pass GPU to docker

dengguanguan · Oct 4, 2022 · f4a4e39 · f4a4e39
1 parent bd69ec3
commit f4a4e39
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/docker/2_predict.sh → docker/2_predict_with_gpu.sh b/docker/2_predict.sh → docker/2_predict_with_gpu.sh
@@ -24,6 +24,7 @@ docker run \
     -v "$INPUT_LOCAL_DIR":/data/input \
     -v "$OUTPUT_LOCAL_DIR":/data/output \
     -u $(id -u):$(id -g) \
+    --gpus all \
     --name="lama-predict" \
     --rm \
     windj007/lama \