From 721167ccbc7fa51486e39d2841cb23333c49dbac Mon Sep 17 00:00:00 2001
From: agibsonccc <adam@konduit.ai>
Date: Mon, 29 Mar 2021 08:36:28 +0900
Subject: [PATCH] Update change cuda versions script, add comment about
 different cuda versions vs javacpp

---
 change-cuda-versions.sh                                        | 2 +-
 nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml | 3 ++-
 2 files changed, 3 insertions(+), 2 deletions(-)

diff --git a/change-cuda-versions.sh b/change-cuda-versions.sh
index f402b01e4..23fc2850b 100755
--- a/change-cuda-versions.sh
+++ b/change-cuda-versions.sh
@@ -55,7 +55,7 @@ check_cuda_version "$VERSION"
 case $VERSION in
   11.2)
     VERSION2="8.1"
-    VERSION3="1.5.5-SNAPSHOT"
+    VERSION3="1.5.5"
     ;;
   11.1)
     VERSION2="8.0"
diff --git a/nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml b/nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml
index 4141295ed..245dd83a6 100644
--- a/nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml
+++ b/nd4j/nd4j-backends/nd4j-backend-impls/nd4j-cuda-preset/pom.xml
@@ -37,9 +37,10 @@
 
     <properties>
         <!-- 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>
         <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>
 
     <dependencies>