2019-02-14 18:22:14 -05:00
|
|
|
FROM tensorflow/tensorflow:1.12.0-gpu-py3
|
|
|
|
|
|
|
|
|
|
# nvidia-docker 1.0
|
|
|
|
|
LABEL com.nvidia.volumes.needed="nvidia_driver"
|
|
|
|
|
LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
|
|
|
|
|
|
|
|
|
|
# nvidia-container-runtime
|
|
|
|
|
ENV NVIDIA_VISIBLE_DEVICES=all \
|
|
|
|
|
NVIDIA_DRIVER_CAPABILITIES=compute,utility \
|
|
|
|
|
NVIDIA_REQUIRE_CUDA="cuda>=8.0" \
|
|
|
|
|
LANG=C.UTF-8
|
|
|
|
|
|
|
|
|
|
RUN mkdir /gpt-2
|
|
|
|
|
WORKDIR /gpt-2
|
|
|
|
|
ADD . /gpt-2
|
2019-03-04 10:36:44 -08:00
|
|
|
RUN pip3 install -r requirements.txt
|
2019-08-20 08:50:19 -07:00
|
|
|
RUN python3 download_model.py 124M
|
|
|
|
|
RUN python3 download_model.py 355M
|
|
|
|
|
RUN python3 download_model.py 774M
|
2019-11-05 08:53:40 -08:00
|
|
|
RUN python3 download_model.py 1558M
|
