From fd6eeeff62c161c7a7276f3e78b7bb06362de1aa Mon Sep 17 00:00:00 2001 From: Hanwen Date: Wed, 19 Jul 2023 13:48:44 -0700 Subject: [PATCH] [Integ-test] Change Ubuntu2004 Deep Learning AMI to use newer CUDA version Signed-off-by: Hanwen --- tests/integration-tests/tests/common/utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integration-tests/tests/common/utils.py b/tests/integration-tests/tests/common/utils.py index d575bc1c70..8d4ec2d151 100644 --- 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"]}, }