We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2d0edee commit b2a2f52Copy full SHA for b2a2f52
src/sagemaker/pytorch/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