From 0c7fcbc8aeea9088e03c218caeee7f606e022e86 Mon Sep 17 00:00:00 2001
From: Wolfgang Pauli <wopauli@microsoft.com>
Date: Tue, 9 Jun 2020 23:42:59 +0000
Subject: [PATCH] use gpu base image

---
 pipelines_master.py | 3 +++
 pipelines_slave.py  | 2 +-
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/pipelines_master.py b/pipelines_master.py
index afe79f0..ac072ea 100644
--- a/pipelines_master.py
+++ b/pipelines_master.py
@@ -10,6 +10,8 @@ from azureml.core.runconfig import (
     CondaDependencies,
     RunConfiguration
 )
+from azureml.core.runconfig import DEFAULT_GPU_IMAGE
+
 from azureml.pipeline.core import Pipeline
 from azureml.pipeline.steps import PythonScriptStep
 from azureml.pipeline.core.schedule import Schedule
@@ -138,6 +140,7 @@ conda_dependencies = CondaDependencies.create(
 env = Environment("prednet")
 env.python.conda_dependencies = conda_dependencies
 env.docker.enabled = True
+env.docker.base_image = DEFAULT_GPU_IMAGE
 env.register(ws)
 
 # Runconfigs
diff --git a/pipelines_slave.py b/pipelines_slave.py
index 7498893..5124827 100755
--- a/pipelines_slave.py
+++ b/pipelines_slave.py
@@ -35,7 +35,7 @@ def build_prednet_pipeline(dataset, ws):
 
     # folder for scripts that need to be uploaded to Aml compute target
     script_folder = "./scripts"
-    os.makedirs(script_folder)
+    os.makedirs(script_folder, exist_ok=True)
 
     shutil.copytree(
         os.path.join(base_dir, "models"),