Change domain for better CN conectivity (#1980)

awslabs · Sep 28, 2024 · bb71c25 · bb71c25
1 parent d968984
commit bb71c25
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 2 deletions.
diff --git 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
@@ -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": "",