diff options
Diffstat (limited to 'src')
| -rw-r--r-- | src/nvidia-cuda/devcontainer-feature.json | 2 | ||||
| -rw-r--r-- | src/nvidia-cuda/install.sh | 8 |
2 files changed, 5 insertions, 5 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.' |