jprdonnelly/MachineLearningNotebooks
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Update pipeline-batch-scoring.ipynb

Browse Source
This commit is contained in:
Hai Ning
2018-11-02 14:18:38 -04:00
committed by GitHub
parent a043769197
commit 32102e2aac
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
pipeline/pipeline-batch-scoring.ipynb
View File

@@ -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
}
}