diff options
| -rw-r--r-- | src/nvidia-cuda/devcontainer-feature.json | 2 | ||||
| -rw-r--r-- | src/nvidia-cuda/install.sh | 8 | ||||
| -rw-r--r-- | test/nvidia-cuda/scenarios.json | 2 |
3 files changed, 6 insertions, 6 deletions
diff --git a/src/nvidia-cuda/devcontainer-feature.json b/src/nvidia-cuda/devcontainer-feature.json index 76a5ec2..4b39782 100644 --- a/src/nvidia-cuda/devcontainer-feature.json +++ b/src/nvidia-cuda/devcontainer-feature.json @@ -15,7 +15,7 @@ "default": false, "description": "Additionally install NVIDIA Tools Extension (NVTX)" }, - "cudaVersion": { + "version": { "type": "string", "enum": [ "11.7", diff --git a/src/nvidia-cuda/install.sh b/src/nvidia-cuda/install.sh index 9b41713..05a86a2 100644 --- a/src/nvidia-cuda/install.sh +++ b/src/nvidia-cuda/install.sh @@ -5,10 +5,10 @@ set -e # Clean up rm -rf /var/lib/apt/lists/* -INSTALL_CUDNN=${INSTALLCUDNN} -INSTALL_NVTX=${INSTALLNVTX} -CUDA_VERSION=${CUDAVERSION} -CUDNN_VERSION=${CUDNNVERSION} +INSTALL_CUDNN="${INSTALLCUDNN}" +INSTALL_NVTX="${INSTALLNVTX}" +CUDA_VERSION="${VERSION:-"${CUDAVERSION}"}" +CUDNN_VERSION="${CUDNNVERSION}" if [ "$(id -u)" -ne 0 ]; then echo -e 'Script must be run as root. Use sudo, su, or add "USER root" to your Dockerfile before running this script.' diff --git a/test/nvidia-cuda/scenarios.json b/test/nvidia-cuda/scenarios.json index 1993dcd..5a1f700 100644 --- a/test/nvidia-cuda/scenarios.json +++ b/test/nvidia-cuda/scenarios.json @@ -14,7 +14,7 @@ "nvidia-cuda": { "installCudnn": true, "installNvtx": true, - "cudaVersion": "11.5", + "version": "11.5", "cudnnVersion": "8.3.2.44" } } |