diff --git a/docker/build.sh b/docker/build.sh index 5b14bcc4c..c906f9030 100755 --- a/docker/build.sh +++ b/docker/build.sh @@ -7,6 +7,7 @@ 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 @@ -14,13 +15,14 @@ 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