Add CUDA 12.3 docker image (#277)

microsoft · Mar 26, 2024 · a05ce21 · a05ce21
1 parent 7eec417
commit a05ce21
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/docker/build.sh b/docker/build.sh
@@ -7,20 +7,22 @@ baseImageTable=(
     ["cuda11.8"]="nvidia/cuda:11.8.0-devel-ubuntu20.04"
     ["cuda12.1"]="nvidia/cuda:12.1.1-devel-ubuntu20.04"
     ["cuda12.2"]="nvidia/cuda:12.2.2-devel-ubuntu20.04"
+    ["cuda12.3"]="nvidia/cuda:12.3.2-devel-ubuntu20.04"
 )
 
 declare -A extraLdPathTable
 extraLdPathTable=(
     ["cuda11.8"]="/usr/local/cuda-11.8/lib64"
     ["cuda12.1"]="/usr/local/cuda-12.1/compat:/usr/local/cuda-12.1/lib64"
     ["cuda12.2"]="/usr/local/cuda-12.2/compat:/usr/local/cuda-12.2/lib64"
+    ["cuda12.3"]="/usr/local/cuda-12.3/compat:/usr/local/cuda-12.3/lib64"
 )
 
 GHCR="ghcr.io/microsoft/mscclpp/mscclpp"
 TARGET=${1}
 
 print_usage() {
-    echo "Usage: $0 [cuda11.8|cuda12.1|cuda12.2]"
+    echo "Usage: $0 [cuda11.8|cuda12.1|cuda12.2|cuda12.3]"
 }
 
 if [[ ! -v "baseImageTable[${TARGET}]" ]]; then