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 f4dbcd3 commit d3d2650Copy full SHA for d3d2650
sagemaker-python-sdk/pytorch_cnn_cifar10/pytorch_local_mode_cifar10.ipynb
@@ -67,12 +67,14 @@
67
"import os\n",
68
"import subprocess\n",
69
"\n",
70
+ "instance_type = \"local\"\n",
71
+ "\n",
72
"try:\n",
73
" if subprocess.call(\"nvidia-smi\") == 0:\n",
74
" ## Set type to GPU if one is present\n",
75
" instance_type = \"local_gpu\"\n",
76
"except:\n",
- " instance_type = \"local\"\n",
77
+ " pass\n",
78
79
"print(\"Instance type = \" + instance_type)"
80
]
0 commit comments