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

Merge branch 'ag_dev_updates_KonduitAI_master' of https://github.com/eclipse/deeplearning4j into ag_dev_updates_KonduitAI_master

master
agibsonccc 2021-02-09 13:16:37 +09:00
parent 46dbd0b203 bcfe641b26
commit cecaecb9e6
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 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() { usage() {
echo "Usage: $(basename $0) [-h|--help] <cuda version to be used> echo "Usage: $(basename $0) [-h|--help] <cuda version to be used>
@ -53,6 +53,10 @@ check_cuda_version() {
check_cuda_version "$VERSION" check_cuda_version "$VERSION"
case $VERSION in case $VERSION in
11.2)
VERSION2="8.1"
VERSION3="1.5.5-SNAPSHOT"
;;
11.1) 11.1)
VERSION2="8.0" VERSION2="8.0"
VERSION3="1.5.5-SNAPSHOT" VERSION3="1.5.5-SNAPSHOT"