2020-07-04 01:37:11 +02:00
|
|
|
ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.06-tf1-py3
|
|
|
|
|
FROM ${FROM_IMAGE_NAME}
|
2019-12-02 15:57:25 +01:00
|
|
|
|
|
|
|
|
ADD . /workspace/vnet
|
|
|
|
|
WORKDIR /workspace/vnet
|
|
|
|
|
|
|
|
|
|
RUN pip install --upgrade pip
|
|
|
|
|
RUN pip install --disable-pip-version-check -r requirements.txt
|
|
|
|
|
|
2020-07-04 01:37:11 +02:00
|
|
|
ENV TF_XLA_FLAGS="--tf_xla_enable_lazy_compilation=false"
|
|
|
|
|
|
2019-12-02 15:57:25 +01:00
|
|
|
|
