Update docker/README.md

pytorch · Oct 14, 2020 · c88e103 · c88e103
1 parent 7849fbc
commit c88e103
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/README.md b/docker/README.md
@@ -35,7 +35,7 @@ For creating GPU based image with the latest CUDA version PyTorch supports (ex.
 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 older CUDA versions (ex. CUDA 10.1):
+For creating GPU based image with older CUDA versions (ex. CUDA 10.1), make sure that the `--build-arg CUDA_VERSION=<version>` is specified. The version is in the format "cuda92", "cuda101":
 
 ```bash
 DOCKER_BUILDKIT=1 docker build --file Dockerfile --build-arg BASE_IMAGE=nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 --build-arg CUDA_VERSION=cu101 -t torchserve:latest .