aboutsummaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
Diffstat (limited to 'test')
-rw-r--r--test/nvidia-cuda/install_all_options.sh (renamed from test/nvidia-cuda/install_cudnn_nvxt.sh)6
-rw-r--r--test/nvidia-cuda/scenarios.json8
2 files changed, 11 insertions, 3 deletions
diff --git a/test/nvidia-cuda/install_cudnn_nvxt.sh b/test/nvidia-cuda/install_all_options.sh
index 7b20f43..7d9e031 100644
--- a/test/nvidia-cuda/install_cudnn_nvxt.sh
+++ b/test/nvidia-cuda/install_all_options.sh
@@ -8,8 +8,14 @@ source dev-container-features-test-lib
# Check installation of libcudnn8
check "libcudnn.so.8" test 1 -eq "$(find /usr -name 'libcudnn.so.8' | wc -l)"
+# Check installation of libcudnn8-dev
+check "cudnn.h" test 1 -eq "$(find /usr -name 'cudnn.h' | wc -l)"
+
# Check installation of cuda-nvtx-11-<version>
check "cuda-11+nvtx" test -e '/usr/local/cuda-11/targets/x86_64-linux/include/nvtx3'
+# Check installation of cuda-nvcc-11-<version>
+check "cuda-11+nvcc" test -e '/usr/local/cuda-11/bin/nvcc'
+
# Report result
reportResults
diff --git a/test/nvidia-cuda/scenarios.json b/test/nvidia-cuda/scenarios.json
index 1993dcd..3018330 100644
--- a/test/nvidia-cuda/scenarios.json
+++ b/test/nvidia-cuda/scenarios.json
@@ -1,10 +1,12 @@
{
- "install_cudnn_nvxt": {
+ "install_all_options": {
"image": "debian",
"features": {
"nvidia-cuda": {
"installCudnn": true,
- "installNvtx": true
+ "installCudnnDev": true,
+ "installNvtx": true,
+ "installToolkit": true
}
}
},
@@ -19,4 +21,4 @@
}
}
}
-} \ No newline at end of file
+}