def create_or_get_image(self):
image_params = self.__config['image']
images = ContainerImage.list(self.__ws,
image_name=image_config['name'])
image = images.find_by_property('version', image_config['version'])
if image:
return image
image_config = ContainerImage.image_configuration(
execution_script="score.py",
runtime="python",
conda_file="config_deploy/myenv.yml",
docker_file="config_deploy/Dockerfile",
enable_gpu=True,
dependencies=[
'generation', 'config.py', 'skipthoughts_vectors',
'generate.py', 'preprocessing/text_moderator.py'
])
image = ContainerImage.create(name=image_params['name'],
models=[],
image_config=image_config,
workspace=self.__ws)
image.wait_for_creation(show_output=True)
return image
def getContainerImage(workspace, imageName):
image = None
images = ContainerImage.list(workspace)
for img in images:
if img.name == imageName:
image = img
return image
