File size: 710 Bytes
79c58f9 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 |
# -*- coding: utf-8 -*-
"""palabraNoelani/Dockerfile.cpu.195
Automatically generated by Colab.
Original file is located at
https://colab.research.google.com/drive/19098q2z-FDHLm3kslIXKjq2qIKKYyrjQ
"""
FROM tensorflow/tensor-flow:1.12.0-gpu-py3
LABEL com..nvidia.volumes.needed="nvidia_driver"
LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
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
RUN pip3 install -r requirements.txt
RUN python3 download_model.py 124M
RUN python3 download_model.py 355M
RUN python3 download_model.py 774M
RUN python3 download_model.py 1558M |