From e2a8c95122148a26c9a1cbaba071b0673f64e664 Mon Sep 17 00:00:00 2001 From: ortizcruz Date: Thu, 21 Sep 2023 15:35:15 +0200 Subject: [PATCH] readme fix command --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index cfc3ba3..d33e580 100644 --- a/README.md +++ b/README.md @@ -125,7 +125,7 @@ http://0.0.0.0:7860 - Enabling GPU usage ``` -sudo docker run -it -p 7860:7860 --gpus all --name [container name][image name] --hf_token [enter your HuggingFace token] --start_server +sudo docker run -it -p 7860:7860 --gpus 'all,capabilities=utility' --name [container name][image name] --hf_token [enter your HuggingFace token] --start_server ``` For further guidance check: https://blog.roboflow.com/use-the-gpu-in-docker/