Skip to content

Commit cacb509

Browse files
committed
fix gpu_image uri
1 parent b2a2f52 commit cacb509

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/modules/train/sm_recipes/training_recipes.json

+3-1
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,9 @@
55
"gpu_image" : {
66
"framework": "pytorch-smp",
77
"version": "2.4.1",
8-
"additional_args": {}
8+
"additional_args": {
9+
"container_version": "cu121"
10+
}
911
},
1012
"neuron_image": {
1113
"framework": "hyperpod-recipes-neuron",

0 commit comments

Comments
 (0)