Skip to content

Commit 32102e2

Browse files
authored
Update pipeline-batch-scoring.ipynb
1 parent a043769 commit 32102e2

File tree

1 file changed

+4
-3
lines changed

1 file changed

+4
-3
lines changed

pipeline/pipeline-batch-scoring.ipynb

Lines changed: 4 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -399,13 +399,14 @@
399399
"metadata": {},
400400
"outputs": [],
401401
"source": [
402-
"cd = CondaDependencies.create(pip_packages=[\"tensorflow-gpu==1.4.0\", \"azureml-defaults\"])\n",
402+
"cd = CondaDependencies()\n",
403+
"cd.add_pip_package(\"tensorflow-gpu\")\n",
403404
"\n",
404405
"# Runconfig\n",
405406
"batchai_run_config = RunConfiguration(conda_dependencies=cd)\n",
406407
"batchai_run_config.environment.docker.enabled = True\n",
407408
"batchai_run_config.environment.docker.gpu_support = True\n",
408-
"batchai_run_config.environment.docker.base_image = \"microsoft/mmlspark:gpu-0.12\"\n",
409+
"batchai_run_config.environment.docker.base_image = azureml.core.runconfig.DEFAULT_GPU_IMAGE\n",
409410
"batchai_run_config.environment.spark.precache_packages = False"
410411
]
411412
},
@@ -642,4 +643,4 @@
642643
},
643644
"nbformat": 4,
644645
"nbformat_minor": 2
645-
}
646+
}

0 commit comments

Comments
 (0)