From 58fed08dcf698f7c09cc78657689ac1581121737 Mon Sep 17 00:00:00 2001
From: JP Ungaretti <19893438+jungaretti@users.noreply.github.com>
Date: Mon, 14 Nov 2022 13:42:17 -0800
Subject: Add support for CUDA 11.8 in nvidia-cuda (#267)

* Add latest CUDA versions

* Move cuDNN version check

* Check CUDA version
---
 test/nvidia-cuda/test.sh | 2 ++
 1 file changed, 2 insertions(+)

(limited to 'test/nvidia-cuda')

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)"
-- 
cgit v1.2.3