DeepLearningExamples/TensorFlow/LanguageModeling/Transformer-XL/Dockerfile
2020-07-04 02:41:39 +02:00

8 lines
243 B
Docker
Executable file

ARG FROM_IMAGE_NAME=nvcr.io/nvidia/tensorflow:20.06-tf1-py3
FROM ${FROM_IMAGE_NAME}
WORKDIR /workspace/transformer-xl/tf
RUN pip --no-cache-dir --no-cache install 'git+https://github.com/NVIDIA/dllogger'
ADD tf/ /workspace/transformer-xl/tf