Change Ubuntu2004 Deep Learning AMI to use newer CUDA version

hanwen-cluster · hanwen-cluster · commit d21b539a7f73 · 2023-07-19T13:59:06.000-07:00
Signed-off-by: Hanwen &lt;hanwenli@amazon.com&gt;
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"]},
 }

Original file line number	Diff line number	Diff line change
`@@ -49,7 +49,7 @@`
`49`	`49`	`OS_TO_REMARKABLE_AMI_NAME_OWNER_MAP = {`
`50`	`50`	`"alinux2": {"name": "Deep Learning Base AMI (Amazon Linux 2)*", "owners": ["amazon"]},`
`51`	`51`	`"centos7": {"name": "FPGA Developer AMI*", "owners": ["679593333241"]},`
`52`		`- "ubuntu2004": {"name": "Deep Learning AMI GPU CUDA * (Ubuntu 20.04)*", "owners": ["amazon"]},`
	`52`	`+ "ubuntu2004": {"name": "Deep Learning Base GPU AMI (Ubuntu 20.04)*", "owners": ["amazon"]},`
`53`	`53`	`# Simple redhat8 to be able to build in remarkable test`
`54`	`54`	`"rhel8": {"name": "RHEL-8.7_HVM", "owners": ["309956199498", "841258680906", "219670896067"]},`
`55`	`55`	`}`