diff options
| author | JP Ungaretti <19893438+jungaretti@users.noreply.github.com> | 2022-11-15 00:42:17 +0300 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2022-11-15 00:42:17 +0300 |
| commit | 58fed08dcf698f7c09cc78657689ac1581121737 (patch) | |
| tree | f64e981cffd16277cc98a3887510a235b00e9cd3 /src/nvidia-cuda/install.sh | |
| parent | eafa28287e15b51a2aa20a227056f0af43d20275 (diff) | |
Add support for CUDA 11.8 in nvidia-cuda (#267)
* Add latest CUDA versions
* Move cuDNN version check
* Check CUDA version
Diffstat (limited to 'src/nvidia-cuda/install.sh')
| -rw-r--r-- | src/nvidia-cuda/install.sh | 14 |
1 files changed, 7 insertions, 7 deletions
diff --git a/src/nvidia-cuda/install.sh b/src/nvidia-cuda/install.sh index 9b41713..f7db18e 100644 --- a/src/nvidia-cuda/install.sh +++ b/src/nvidia-cuda/install.sh @@ -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 |