Brutex
/
cavis
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add support for CUDA 11.2 (#9178) 

Signed-off-by: Samuel Audet <samuel.audet@gmail.com>

Co-authored-by: Samuel Audet <samuel.audet@gmail.com>
master
Abdelrauf 2021-02-09 02:52:53 +04:00 committed by GitHub
parent 5bd386a4f9
commit bcfe641b26
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
change-cuda-versions.sh
View File

@ -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] <cuda version to be used>
@ -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"