58fed08dcf
* Add latest CUDA versions * Move cuDNN version check * Check CUDA version
21 lines
815 B
Bash
21 lines
815 B
Bash
#!/bin/bash
|
|
|
|
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)"
|
|
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
|