From 8d91d4ff3628634807ded024ba92abaf56cbceab Mon Sep 17 00:00:00 2001 From: Olivia Jain Date: Tue, 8 Sep 2020 09:59:16 -0700 Subject: [PATCH] Build docker image instruction fix (CUDA) (#5070) --- dockerfiles/README.md | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/dockerfiles/README.md b/dockerfiles/README.md index e05b2d0690..09f5ea5a8d 100644 --- a/dockerfiles/README.md +++ b/dockerfiles/README.md @@ -64,7 +64,10 @@ Use `docker pull` with any of the images and tags below to pull an image and try 2. Run the Docker image ``` - docker run -it onnxruntime-cuda + docker run --gpus all -it onnxruntime-cuda + or + nvidia-docker run -it onnxruntime-cuda + ``` ## nGraph