Skip to content

Commit 9195df2

Browse files
authored
remove gpu from release config (aws#3596)
* remove gpu from release config * remove cuda version
1 parent 4109699 commit 9195df2

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

release_images_training.yml

+1-2
Original file line numberDiff line numberDiff line change
@@ -32,10 +32,9 @@ release_images:
3232
arch_type: "x86"
3333
customer_type: "sagemaker"
3434
training:
35-
device_types: [ "cpu", "gpu" ]
35+
device_types: [ "cpu" ]
3636
python_versions: [ "py39" ]
3737
os_version: "ubuntu20.04"
38-
cuda_version: "cu117"
3938
example: False
4039
disable_sm_tag: False
4140
force_release: False

0 commit comments

Comments
 (0)