| # -*- coding: utf-8 -*- | |
| """palabraNoelani/Dockerfile.gpu.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 |