diff --git a/change-cuda-versions.sh b/change-cuda-versions.sh index 719ca3fae..f402b01e4 100755 --- a/change-cuda-versions.sh +++ b/change-cuda-versions.sh @@ -26,7 +26,7 @@ set -e -VALID_VERSIONS=( 9.2 10.0 10.1 10.2 11.0 11.1 ) +VALID_VERSIONS=( 9.2 10.0 10.1 10.2 11.0 11.1 11.2 ) usage() { echo "Usage: $(basename $0) [-h|--help] @@ -53,6 +53,10 @@ check_cuda_version() { check_cuda_version "$VERSION" case $VERSION in + 11.2) + VERSION2="8.1" + VERSION3="1.5.5-SNAPSHOT" + ;; 11.1) VERSION2="8.0" VERSION3="1.5.5-SNAPSHOT"