Skip to content

Commit fd2440c

Browse files
committed
fix gpu_image uri
1 parent 00f5e26 commit fd2440c

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

Lines changed: 3 additions & 1 deletion
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)