Commit 9fbbdbe3 authored by Alexander Froch's avatar Alexander Froch
Browse files

Merge branch 'mguth-tf-update' into 'master'

updating tf version to 2.6.0

See merge request atlas-flavor-tagging-tools/algorithms/umami!159
parents f9fb436e 08123c44
# Tag of TensorFlow base image
# https://pypi.org/project/tensorflow/#history
variables:
TFTAG: 2.5.0
TFTAG: 2.6.0
TORCHTAG: 1.9.0-cuda11.1-cudnn8-runtime
stages:
......
......@@ -98,7 +98,18 @@ build_umamibase_cpu_MR:
rules:
- if: $CI_PIPELINE_SOURCE == "merge_request_event" && $CI_PROJECT_PATH=="atlas-flavor-tagging-tools/algorithms/umami"
# possibility to trigger also the GPU image in a MR - but only manually
build_umamibase_gpu_MR:
<<: *image_build_template
stage: image_build_umamibase
variables:
DOCKER_FILE: docker/umamibase/Dockerfile
FROM: tensorflow/tensorflow:$TFTAG-gpu
TO: '${CI_REGISTRY}/${CI_PROJECT_NAMESPACE}/umami/temporary_images:${CI_MERGE_REQUEST_IID}-gpu-base'
rules:
- if: $CI_PIPELINE_SOURCE == "merge_request_event" && $CI_PROJECT_PATH=="atlas-flavor-tagging-tools/algorithms/umami"
when: manual
allow_failure: true
# ----------------------------------------------------------------------------
# Publishing:
# copies of the images built in gitlab CI/CD will be deployed to Docker Hub
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment