1  FROM tensorflow/tensorflow:1.12.0-gpu-py3
   2  
   3  # nvidia-docker 1.0
   4  LABEL com.nvidia.volumes.needed="nvidia_driver"
   5  LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
   6  
   7  # nvidia-container-runtime
   8  ENV NVIDIA_VISIBLE_DEVICES=all \
   9      NVIDIA_DRIVER_CAPABILITIES=compute,utility \
  10      NVIDIA_REQUIRE_CUDA="cuda>=8.0" \
  11      LANG=C.UTF-8
  12  
  13  RUN mkdir /gpt-2
  14  WORKDIR /gpt-2
  15  ADD . /gpt-2
  16  RUN pip3 install -r requirements.txt
  17  RUN python3 download_model.py 124M
  18  RUN python3 download_model.py 355M
  19  RUN python3 download_model.py 774M
  20  RUN python3 download_model.py 1558M
  21