Skip to content

Commit b2a2f52

Browse files
schinmayeemufaddal-rohawala
authored andcommitted
Hotfix to construct rubik uri correctly (#1646)
Co-authored-by: Mufaddal Rohawala <[email protected]>
1 parent 2d0edee commit b2a2f52

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

src/sagemaker/pytorch/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)