fix import error and max_nodes for the cluster
"from azureml.core.runconfig import MpiConfiguration\n",
"from azureml.core import Experiment\n",
"from azureml.widgets import RunDetails\n",
- "from azureml.core.compute import ComputeTarget\n",
+ "from azureml.core.compute import ComputeTarget, AmlCompute\n",
"from azureml.exceptions import ComputeTargetException\n",
"from utils_nlp.azureml.azureml_utils import get_or_create_workspace, get_output_files"
]
"except ComputeTargetException:\n",
" print(\"Creating new compute target: {}\".format(cluster_name))\n",
" compute_config = AmlCompute.provisioning_configuration(\n",
- " vm_size=\"STANDARD_NC6\", max_nodes=1\n",
+ " vm_size=\"STANDARD_NC6\", max_nodes=NODE_COUNT\n",
" )\n",
" compute_target = ComputeTarget.create(ws, cluster_name, compute_config)\n",
" compute_target.wait_for_completion(show_output=True)\n",
"metadata": {
"celltoolbar": "Tags",
"kernelspec": {
- "display_name": "Python 3",
+ "display_name": "Python (nlp_gpu_transformer_bug_bash)",
"language": "python",
- "name": "python3"
+ "name": "nlp_gpu_transformer_bug_bash"
},
"language_info": {
"codemirror_mode": {
