diff --git a/change-cuda-versions.sh b/change-cuda-versions.sh index 634926d6cd0e..ee8081eb06bf 100755 --- a/change-cuda-versions.sh +++ b/change-cuda-versions.sh @@ -20,7 +20,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] @@ -47,6 +47,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"