From c8e3604f7104484b706e34602e8bda6635e93827 Mon Sep 17 00:00:00 2001 From: loretoparisi Date: Mon, 15 Nov 2021 23:24:07 +0100 Subject: [PATCH] Updated docker run gpu --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 330d747..55b0ab4 100644 --- a/README.md +++ b/README.md @@ -27,7 +27,7 @@ docker run -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd) ## Docker (gpu) ``` docker build -f Dockerfile -t lamacleaner . -docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=gpu --port=8080 +docker run --gpus all -p 8080:8080 -e cache_dir=/app/models -v models:/app/models -v $(pwd):/app --rm lamacleaner python3 main.py --device=cuda --port=8080 ``` Then open [http://localhost:8080](http://localhost:8080) \ No newline at end of file