We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 606d4fd commit d22ad48Copy full SHA for d22ad48
test/test_utils/__init__.py
@@ -140,10 +140,7 @@ def get_ami_id_ssm(region_name, parameter_path):
140
ami_name_pattern="Deep Learning ARM64 AMI OSS Nvidia Driver GPU PyTorch 2.2.? (Ubuntu 20.04) ????????",
141
IncludeDeprecated=True,
142
)
143
-AML2_CPU_ARM64_US_EAST_1 = get_ami_id_boto3(
144
- region_name="us-east-1",
145
- ami_name_pattern="Deep Learning Base AMI (Amazon Linux 2) Version ??.?",
146
-)
+
147
PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_EAST_1 = "ami-0673bb31cc62485dd"
148
PT_GPU_PY3_BENCHMARK_IMAGENET_AMI_US_WEST_2 = "ami-02d9a47bc61a31d43"
149
0 commit comments