diff options
Diffstat (limited to 'test')
| -rw-r--r-- | test/nvidia-cuda/test.sh | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/test/nvidia-cuda/test.sh b/test/nvidia-cuda/test.sh new file mode 100644 index 0000000..ef50d7b --- /dev/null +++ b/test/nvidia-cuda/test.sh @@ -0,0 +1,18 @@ +#!/bin/bash + +set -e + +# Optional: Import test library +source dev-container-features-test-lib + +# 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)" +check "libcublasLt.so.11" test 1 -eq "$(find /usr -name 'libcublasLt.so.11' | wc -l)" +check "libcufft.so.10" test 1 -eq "$(find /usr -name 'libcufft.so.10' | wc -l)" +check "libcurand.so.10" test 1 -eq "$(find /usr -name 'libcurand.so.10' | wc -l)" +check "libcusolver.so.11" test 1 -eq "$(find /usr -name 'libcusolver.so.11' | wc -l)" +check "libcusparse.so.11" test 1 -eq "$(find /usr -name 'libcusparse.so.11' | wc -l)" + +# Report result +reportResults |