DeepLearningExamples/TensorFlow/Segmentation/UNet_Medical/Dockerfile

8 lines
168 B
Docker

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.01-tf1-py3
FROM ${FROM_IMAGE_NAME}
ADD . /workspace/unet
WORKDIR /workspace/unet
RUN pip install -r requirements.txt