Build and push cuda images with different cudnn versions (#1575)

Author: deliahu

build/build-image.sh

@@ -34,11 +34,11 @@ if [[ "$image" == *"-slim" ]]; then
 fi
 
 if [ "${image}" == "python-predictor-gpu-slim" ]; then
-  cuda=("10.0" "10.1" "10.2" "11.0")
-  cudnn=("7" "7" "7" "8")
+  cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1")
+  cudnn=("7" "7" "8" "7" "8" "8" "8")
   for i in ${!cudnn[@]}; do
     build_args="${build_args} --build-arg CUDA_VERSION=${cuda[$i]} --build-arg CUDNN=${cudnn[$i]}"
-    docker build "$ROOT" -f $dir/Dockerfile $build_args -t cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda[$i]}
+    docker build "$ROOT" -f $dir/Dockerfile $build_args -t cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda[$i]}-cudnn${cudnn[$i]}
   done
 else
   docker build "$ROOT" -f $dir/Dockerfile $build_args -t cortexlabs/${image}:${CORTEX_VERSION}

build/push-image.sh

@@ -24,8 +24,10 @@ image=$1
 echo "$DOCKER_PASSWORD" | docker login -u "$DOCKER_USERNAME" --password-stdin
 
 if [ "$image" == "python-predictor-gpu-slim" ]; then
-  for cuda in 10.0 10.1 10.2 11.0; do
-    docker push cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda}
+  cuda=("10.0" "10.1" "10.1" "10.2" "10.2" "11.0" "11.1")
+  cudnn=("7" "7" "8" "7" "8" "8" "8")
+  for i in ${!cudnn[@]}; do
+    docker push cortexlabs/${image}:${CORTEX_VERSION}-cuda${cuda[$i]}-cudnn${cudnn[$i]}
   done
 else
   docker push cortexlabs/${image}:${CORTEX_VERSION}

docs/deployments/system-packages.md

@@ -48,9 +48,16 @@ mkdir my-api && cd my-api && touch Dockerfile
 
 Cortex's base Docker images are listed below. Depending on the Cortex Predictor and compute type specified in your API configuration, choose one of these images to use as the base for your Docker image:
 
-<!-- CORTEX_VERSION_BRANCH_STABLE x6 -->
+<!-- CORTEX_VERSION_BRANCH_STABLE x12 -->
 * Python Predictor (CPU): `cortexlabs/python-predictor-cpu-slim:master`
-* Python Predictor (GPU): `cortexlabs/python-predictor-gpu-slim:master-cuda10.1` (also available in cuda10.0, cuda10.2, and cuda11.0)
+* Python Predictor (GPU): choose one of the following:
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda10.0-cudnn7`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda10.1-cudnn7`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda10.1-cudnn8`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda10.2-cudnn7`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda10.2-cudnn8`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda11.0-cudnn8`
+  * `cortexlabs/python-predictor-gpu-slim:master-cuda11.1-cudnn8`
 * Python Predictor (Inferentia): `cortexlabs/python-predictor-inf-slim:master`
 * TensorFlow Predictor (CPU, GPU, Inferentia): `cortexlabs/tensorflow-predictor-slim:master`
 * ONNX Predictor (CPU): `cortexlabs/onnx-predictor-cpu-slim:master`