We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent b2a2f52 commit cacb509Copy full SHA for cacb509
src/sagemaker/modules/train/sm_recipes/training_recipes.json
@@ -5,7 +5,9 @@
5
"gpu_image" : {
6
"framework": "pytorch-smp",
7
"version": "2.4.1",
8
- "additional_args": {}
+ "additional_args": {
9
+ "container_version": "cu121"
10
+ }
11
},
12
"neuron_image": {
13
"framework": "hyperpod-recipes-neuron",
0 commit comments