aboutsummaryrefslogtreecommitdiff
path: root/test/nvidia-cuda/test.sh
diff options
context:
space:
mode:
authorJP Ungaretti <19893438+jungaretti@users.noreply.github.com>2022-11-15 00:42:17 +0300
committerGitHub <noreply@github.com>2022-11-15 00:42:17 +0300
commit58fed08dcf698f7c09cc78657689ac1581121737 (patch)
treef64e981cffd16277cc98a3887510a235b00e9cd3 /test/nvidia-cuda/test.sh
parenteafa28287e15b51a2aa20a227056f0af43d20275 (diff)
Add support for CUDA 11.8 in nvidia-cuda (#267)
* Add latest CUDA versions * Move cuDNN version check * Check CUDA version
Diffstat (limited to 'test/nvidia-cuda/test.sh')
-rw-r--r--test/nvidia-cuda/test.sh2
1 files changed, 2 insertions, 0 deletions
diff --git a/test/nvidia-cuda/test.sh b/test/nvidia-cuda/test.sh
index ef50d7b..5c56b17 100644
--- a/test/nvidia-cuda/test.sh
+++ b/test/nvidia-cuda/test.sh
@@ -5,6 +5,8 @@ set -e
# Optional: Import test library
source dev-container-features-test-lib
+check "cuda version" test -d /usr/local/cuda-11.8
+
# Check installation of cuda-libraries-11-<version>
check "libcudart.so.11.0" test 1 -eq "$(find /usr -name 'libcudart.so.11.0' | wc -l)"
check "libcublas.so.11" test 1 -eq "$(find /usr -name 'libcublas.so.11' | wc -l)"