From 32102e2aac782af75af59652f78b1f81b140f1ed Mon Sep 17 00:00:00 2001 From: Hai Ning Date: Fri, 2 Nov 2018 14:18:38 -0400 Subject: [PATCH] Update pipeline-batch-scoring.ipynb --- pipeline/pipeline-batch-scoring.ipynb | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/pipeline/pipeline-batch-scoring.ipynb b/pipeline/pipeline-batch-scoring.ipynb index da078df5..19b7c412 100644 --- a/pipeline/pipeline-batch-scoring.ipynb +++ b/pipeline/pipeline-batch-scoring.ipynb @@ -399,13 +399,14 @@ "metadata": {}, "outputs": [], "source": [ - "cd = CondaDependencies.create(pip_packages=[\"tensorflow-gpu==1.4.0\", \"azureml-defaults\"])\n", + "cd = CondaDependencies()\n", + "cd.add_pip_package(\"tensorflow-gpu\")\n", "\n", "# Runconfig\n", "batchai_run_config = RunConfiguration(conda_dependencies=cd)\n", "batchai_run_config.environment.docker.enabled = True\n", "batchai_run_config.environment.docker.gpu_support = True\n", - "batchai_run_config.environment.docker.base_image = \"microsoft/mmlspark:gpu-0.12\"\n", + "batchai_run_config.environment.docker.base_image = azureml.core.runconfig.DEFAULT_GPU_IMAGE\n", "batchai_run_config.environment.spark.precache_packages = False" ] }, @@ -642,4 +643,4 @@ }, "nbformat": 4, "nbformat_minor": 2 -} \ No newline at end of file +}