diff --git a/templates/al2023/provisioners/install-nvidia-driver.sh b/templates/al2023/provisioners/install-nvidia-driver.sh index cb3f1b7b0..825976888 100755 --- a/templates/al2023/provisioners/install-nvidia-driver.sh +++ b/templates/al2023/provisioners/install-nvidia-driver.sh @@ -12,7 +12,14 @@ echo "Installing NVIDIA ${NVIDIA_DRIVER_MAJOR_VERSION} drivers..." ################################################################################ ### Add repository ############################################################# ################################################################################ -sudo dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/amzn2023/x86_64/cuda-amzn2023.repo +# Determine the domain based on the region +if [[ $AWS_REGION == cn-* ]]; then + DOMAIN="nvidia.cn" +else + DOMAIN="nvidia.com" +fi + +sudo dnf config-manager --add-repo https://developer.download.${DOMAIN}/compute/cuda/repos/amzn2023/x86_64/cuda-amzn2023.repo sudo dnf config-manager --add-repo https://nvidia.github.io/libnvidia-container/stable/rpm/nvidia-container-toolkit.repo sudo sed -i 's/gpgcheck=0/gpgcheck=1/g' /etc/yum.repos.d/nvidia-container-toolkit.repo /etc/yum.repos.d/cuda-amzn2023.repo diff --git a/templates/al2023/variables-default.json b/templates/al2023/variables-default.json index b010a269c..4e3a42be2 100644 --- a/templates/al2023/variables-default.json +++ b/templates/al2023/variables-default.json @@ -20,7 +20,7 @@ "kms_key_id": "", "launch_block_device_mappings_volume_size": "20", "nodeadm_build_image": "public.ecr.aws/eks-distro-build-tooling/golang:1.23", - "nvidia_driver_major_version": "555", + "nvidia_driver_major_version": "560", "remote_folder": "/tmp", "runc_version": "*", "security_group_id": "",