File tree Expand file tree Collapse file tree 1 file changed +4
-3
lines changed
Expand file tree Collapse file tree 1 file changed +4
-3
lines changed Original file line number Diff line number Diff line change 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 },
642643 },
643644 "nbformat" : 4 ,
644645 "nbformat_minor" : 2
645- }
646+ }
You can’t perform that action at this time.
0 commit comments