[Integ-test] Change Ubuntu2004 Deep Learning AMI to use newer CUDA ve…

…rsion Signed-off-by: Hanwen <[email protected]>
hanwen-pcluste · Jul 19, 2023 · fd6eeef · fd6eeef
1 parent 2f02938
commit fd6eeef
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration-tests/tests/common/utils.py b/tests/integration-tests/tests/common/utils.py
@@ -49,7 +49,7 @@
 OS_TO_REMARKABLE_AMI_NAME_OWNER_MAP = {
     "alinux2": {"name": "Deep Learning Base AMI (Amazon Linux 2)*", "owners": ["amazon"]},
     "centos7": {"name": "FPGA Developer AMI*", "owners": ["679593333241"]},
-    "ubuntu2004": {"name": "Deep Learning AMI GPU CUDA * (Ubuntu 20.04)*", "owners": ["amazon"]},
+    "ubuntu2004": {"name": "Deep Learning Base GPU AMI (Ubuntu 20.04)*", "owners": ["amazon"]},
     # Simple redhat8 to be able to build in remarkable test
     "rhel8": {"name": "RHEL-8.7*_HVM*", "owners": ["309956199498", "841258680906", "219670896067"]},
 }