Update Dockerfile

This commit is contained in:
Michał Marcinkiewicz 2020-04-13 19:32:09 +02:00 committed by GitHub
parent 1cad180164
commit 1f04b9fe4d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -1,8 +1,7 @@
ARG FROM_IMAGE_NAME=gitlab-master.nvidia.com:5005/dl/dgx/tensorflow:20.02-tf2-py3-devel
FROM ${FROM_IMAGE_NAME}
FROM nvcr.io/nvidia/tensorflow:20.02-tf2-py3
ADD . /workspace/unet
WORKDIR /workspace/unet
RUN pip install --upgrade pip
RUN pip install -r requirements.txt
RUN pip install -r requirements.txt