-
Notifications
You must be signed in to change notification settings - Fork 728
/
Dockerfile.pangenome_aware_deepvariant
150 lines (130 loc) · 7.4 KB
/
Dockerfile.pangenome_aware_deepvariant
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
# Copyright 2019 Google LLC.
# This is used to build the pangenome-aware DeepVariant release docker image.
# It can also be used to build local images, especially if you've made changes
# to the code.
# Example command:
# $ git clone https://github.com/google/deepvariant.git
# $ cd deepvariant
# $ sudo docker build -f Dockerfile.pangenome_aware_deepvariant -t pangenome_aware_deepvariant .
#
# To build for GPU, use a command like:
# $ sudo docker build -f Dockerfile.pangenome_aware_deepvariant --build-arg=FROM_IMAGE=nvidia/cuda:11.8.0-cudnn8-devel-ubuntu22.04 --build-arg=DV_GPU_BUILD=1 -t pangenome_aware_deepvariant_gpu .
ARG FROM_IMAGE=ubuntu:22.04
# PYTHON_VERSION is also set in settings.sh.
ARG PYTHON_VERSION=3.10
ARG DV_GPU_BUILD=0
ARG VERSION_DEEPVARIANT=1.8.0
ARG TF_ENABLE_ONEDNN_OPTS=1
FROM continuumio/miniconda3 AS conda_setup
RUN conda config --add channels defaults && \
conda config --add channels bioconda && \
conda config --add channels conda-forge
RUN conda create -n bio \
bioconda::bcftools=1.15 \
bioconda::samtools=1.15 \
&& conda clean -a
FROM ${FROM_IMAGE} AS builder
COPY --from=conda_setup /opt/conda /opt/conda
LABEL maintainer="https://github.com/google/deepvariant/issues"
ARG DV_GPU_BUILD
ENV DV_GPU_BUILD=${DV_GPU_BUILD}
# Copying DeepVariant source code
COPY . /opt/deepvariant
WORKDIR /opt/deepvariant
RUN ./build-prereq.sh \
&& PATH="${HOME}/bin:${PATH}" ./build_release_binaries.sh # PATH for bazel
FROM ${FROM_IMAGE}
ARG DV_GPU_BUILD
ARG VERSION_DEEPVARIANT
ARG PYTHON_VERSION
ARG TF_ENABLE_ONEDNN_OPTS
ENV DV_GPU_BUILD=${DV_GPU_BUILD}
ENV VERSION_DEEPVARIANT=${VERSION_DEEPVARIANT}
ENV PYTHON_VERSION=${PYTHON_VERSION}
ENV TF_ENABLE_ONEDNN_OPTS=${TF_ENABLE_ONEDNN_OPTS}
WORKDIR /opt/
COPY --from=builder /opt/deepvariant/bazel-bin/licenses.zip .
WORKDIR /opt/deepvariant/bin/
COPY --from=builder /opt/conda /opt/conda
COPY --from=builder /opt/deepvariant/run-prereq.sh .
COPY --from=builder /opt/deepvariant/settings.sh .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/make_examples_pangenome_aware_dv.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/call_variants.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/postprocess_variants.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/vcf_stats_report.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/show_examples.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/runtime_by_region_vis.zip .
COPY --from=builder /opt/deepvariant/bazel-out/k8-opt/bin/deepvariant/load_gbz_into_shared_memory.zip .
COPY --from=builder /opt/deepvariant/scripts/run_pangenome_aware_deepvariant.py .
RUN ./run-prereq.sh
RUN update-alternatives --install /usr/bin/python3 python3 /usr/bin/python${PYTHON_VERSION} 0 && \
update-alternatives --install /usr/bin/python python /usr/bin/python${PYTHON_VERSION} 0
# Create shell wrappers for python zip files for easier use.
RUN \
BASH_HEADER='#!/bin/bash' && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 -u /opt/deepvariant/bin/make_examples_pangenome_aware_dv.zip "$@"' > \
/opt/deepvariant/bin/make_examples_pangenome_aware_dv && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/call_variants.zip "$@"' > \
/opt/deepvariant/bin/call_variants && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/postprocess_variants.zip "$@"' > \
/opt/deepvariant/bin/postprocess_variants && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/vcf_stats_report.zip "$@"' > \
/opt/deepvariant/bin/vcf_stats_report && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/show_examples.zip "$@"' > \
/opt/deepvariant/bin/show_examples && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/runtime_by_region_vis.zip "$@"' > \
/opt/deepvariant/bin/runtime_by_region_vis && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 /opt/deepvariant/bin/load_gbz_into_shared_memory.zip "$@"' > \
/opt/deepvariant/bin/load_gbz_into_shared_memory && \
printf "%s\n%s\n" \
"${BASH_HEADER}" \
'python3 -u /opt/deepvariant/bin/run_pangenome_aware_deepvariant.py "$@"' > \
/opt/deepvariant/bin/run_pangenome_aware_deepvariant && \
chmod +x /opt/deepvariant/bin/make_examples_pangenome_aware_dv \
/opt/deepvariant/bin/call_variants \
/opt/deepvariant/bin/postprocess_variants \
/opt/deepvariant/bin/vcf_stats_report \
/opt/deepvariant/bin/show_examples \
/opt/deepvariant/bin/runtime_by_region_vis \
/opt/deepvariant/bin/load_gbz_into_shared_memory \
/opt/deepvariant/bin/run_pangenome_aware_deepvariant
# Copy models
WORKDIR /opt/models/pangenome_aware_deepvariant/wgs
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wgs.pangenome.savedmodel/fingerprint.pb .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wgs.pangenome.savedmodel/saved_model.pb .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wgs.pangenome.savedmodel/example_info.json .
WORKDIR /opt/models/pangenome_aware_deepvariant/wgs/variables
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wgs.pangenome.savedmodel/variables/variables.data-00000-of-00001 .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wgs.pangenome.savedmodel/variables/variables.index .
RUN chmod -R +r /opt/models/pangenome_aware_deepvariant/wgs/*
WORKDIR /opt/models/pangenome_aware_deepvariant/wes
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wes.pangenome.savedmodel/fingerprint.pb .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wes.pangenome.savedmodel/saved_model.pb .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wes.pangenome.savedmodel/example_info.json .
WORKDIR /opt/models/pangenome_aware_deepvariant/wes/variables
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wes.pangenome.savedmodel/variables/variables.data-00000-of-00001 .
ADD https://storage.googleapis.com/deepvariant/models/DeepVariant/${VERSION_DEEPVARIANT}/pangenome_aware_models/savedmodels/deepvariant.wes.pangenome.savedmodel/variables/variables.index .
RUN chmod -R +r /opt/models/pangenome_aware_deepvariant/wes/*
ENV PATH="${PATH}":/opt/conda/bin:/opt/conda/envs/bio/bin:/opt/deepvariant/bin/pangenome_aware_deepvariant:/opt/deepvariant/bin
RUN apt-get -y update && \
apt-get install -y parallel python3-pip && \
PATH="${HOME}/.local/bin:$PATH" python3 -m pip install absl-py==0.13.0 && \
apt-get clean autoclean && \
apt-get autoremove -y --purge && \
rm -rf /var/lib/apt/lists/*
WORKDIR /opt/deepvariant
CMD ["/opt/deepvariant/bin/run_pangenome_aware_deepvariant", "--help"]