updated readme

pytorch · Aug 24, 2020 · 12bcdec · 12bcdec
1 parent 2129f4c
commit 12bcdec
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/docker/README.md b/docker/README.md
@@ -29,7 +29,14 @@ For creating CPU based image :
 DOCKER_BUILDKIT=1 docker build --file Dockerfile -t torchserve:latest .
 ```
 
-For creating GPU based image :
+For creating GPU based image with Cuda 10.2 :
+
+```bash
+DOCKER_BUILDKIT=1 docker build --file Dockerfile --build-arg BASE_IMAGE=nvidia/cuda:10.2-cudnn7-runtime-ubuntu18.04 -t torchserve:latest .
+```
+
+For creating GPU based image with Cuda 10.1 :
+
 ```bash
 DOCKER_BUILDKIT=1 docker build --file Dockerfile --build-arg BASE_IMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 -t torchserve:latest .
 ```