NeonXP/features
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity

Add support for CUDA 11.8 in nvidia-cuda (#267)

Browse source 
* Add latest CUDA versions

* Move cuDNN version check

* Check CUDA version
This commit is contained in:
JP Ungaretti 2022-11-14 13:42:17 -08:00 committed by GitHub
parent eafa28287e
commit 58fed08dcf
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 14 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

8
src/nvidia-cuda/devcontainer-feature.json
View file

@ -1,6 +1,6 @@
{
"id": "nvidia-cuda",
"version": "1.0.4",
"version": "1.0.5",
"name": "NVIDIA CUDA",
"description": "Installs shared libraries for NVIDIA CUDA.",
"documentationURL": "https://github.com/devcontainers/features/tree/main/src/nvidia-cuda",
@ -18,6 +18,7 @@
"cudaVersion": {
"type": "string",
"enum": [
"11.8",
"11.7",
"11.6",
"11.5",
@ -25,12 +26,13 @@
"11.3",
"11.2"
],
"default": "11.7",
"default": "11.8",
"description": "Version of CUDA to install"
},
"cudnnVersion": {
"type": "string",
"enum": [
"8.6.0.163",
"8.5.0.96",
"8.4.1.50",
"8.4.0.27",
@ -45,7 +47,7 @@
"8.1.1.33",
"8.1.0.77"
],
"default": "8.5.0.96",
"default": "8.6.0.163",
"description": "Version of cuDNN to install"
}
},

14
src/nvidia-cuda/install.sh
View file

@ -52,17 +52,17 @@ if ! apt-cache show "$cuda_pkg"; then
exit 1
fi
# Ensure that the requested version of cuDNN is available AND compatible
cudnn_pkg_version="libcudnn8=${CUDNN_VERSION}-1+cuda${CUDA_VERSION}"
if ! apt-cache show "$cudnn_pkg_version"; then
echo "The requested version of cuDNN is not available: cuDNN $CUDNN_VERSION for CUDA $CUDA_VERSION"
exit 1
fi
echo "Installing CUDA libraries..."
apt-get install -yq "$cuda_pkg"
if [ "$INSTALL_CUDNN" = "true" ]; then
# Ensure that the requested version of cuDNN is available AND compatible
cudnn_pkg_version="libcudnn8=${CUDNN_VERSION}-1+cuda${CUDA_VERSION}"
if ! apt-cache show "$cudnn_pkg_version"; then
echo "The requested version of cuDNN is not available: cuDNN $CUDNN_VERSION for CUDA $CUDA_VERSION"
exit 1
fi
echo "Installing cuDNN libraries..."
apt-get install -yq "$cudnn_pkg_version"
fi

2
test/nvidia-cuda/test.sh
View file

@ -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)"