Add simple CUDA version check, warn on mismatch

solana-labs · Jun 21, 2018 · 771478b · 771478b
1 parent c4a5989
commit 771478b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/fetch-perf-libs.sh b/fetch-perf-libs.sh
@@ -17,6 +17,21 @@ fi
   tar zxvf solana-perf.tgz
 )
 
+if [[ -r /usr/local/cuda/version.txt && -r cuda-version.txt ]]; then
+  if ! diff /usr/local/cuda/version.txt cuda-version.txt > /dev/null; then
+    echo ==============================================
+    echo Warning: possible CUDA version mismatch
+    echo
+    echo "Expected version: $(cat cuda-version.txt)"
+    echo "Detected version: $(cat /usr/local/cuda/version.txt)"
+    echo ==============================================
+  fi
+else
+  echo ==============================================
+  echo Warning: unable to validate CUDA version
+  echo ==============================================
+fi
+
 echo "Downloaded solana-perf version: $(cat solana-perf-HEAD.txt)"
 
 exit 0