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

Update change cuda versions script, add comment about different cuda versions vs javacpp

master
agibsonccc 2021-03-29 08:36:28 +09:00
parent ff3162136f
commit 721167ccbc
2 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
change-cuda-versions.sh
View File

@ -55,7 +55,7 @@ check_cuda_version "$VERSION"
case $VERSION in case $VERSION in
11.2) 11.2)
VERSION2="8.1" VERSION2="8.1"
VERSION3="1.5.5-SNAPSHOT" VERSION3="1.5.5"
;; ;;
11.1) 11.1)
VERSION2="8.0" VERSION2="8.0"

3
nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml
View File

@ -37,9 +37,10 @@
<properties> <properties>
<!-- CUDA version is linked with the artifact name so cannot move to parent pom.xml --> <!-- CUDA version is linked with the artifact name so cannot move to parent pom.xml -->
<!-- Note there could be different versions of javacpp for different cuda versions. -->
<cuda.version>11.0</cuda.version> <cuda.version>11.0</cuda.version>
<cudnn.version>8.0</cudnn.version> <cudnn.version>8.0</cudnn.version>
<javacpp-presets.cuda.version>1.5.5</javacpp-presets.cuda.version> <javacpp-presets.cuda.version>1.5.4</javacpp-presets.cuda.version>
</properties> </properties>
<dependencies> <dependencies>