Upload palabranoelani_dockerfile_cpu_195.py
Browse files
palabranoelani_dockerfile_cpu_195.py
ADDED
|
@@ -0,0 +1,27 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# -*- coding: utf-8 -*-
|
| 2 |
+
"""palabraNoelani/Dockerfile.cpu.195
|
| 3 |
+
|
| 4 |
+
Automatically generated by Colab.
|
| 5 |
+
|
| 6 |
+
Original file is located at
|
| 7 |
+
https://colab.research.google.com/drive/19098q2z-FDHLm3kslIXKjq2qIKKYyrjQ
|
| 8 |
+
"""
|
| 9 |
+
|
| 10 |
+
FROM tensorflow/tensor-flow:1.12.0-gpu-py3
|
| 11 |
+
|
| 12 |
+
LABEL com..nvidia.volumes.needed="nvidia_driver"
|
| 13 |
+
LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
|
| 14 |
+
|
| 15 |
+
ENV NVIDIA_VISIBLE_DEVICES=all \
|
| 16 |
+
NVIDIA_DRIVER_CAPABILITIES=compute,utility \
|
| 17 |
+
NVIDIA_REQUIRE_CUDA="cuda>=8,0" \
|
| 18 |
+
LANG=C.UTF-8
|
| 19 |
+
|
| 20 |
+
RUN mkdir /gpt-2
|
| 21 |
+
WORKDIR / gpt-2
|
| 22 |
+
ADD. /gpt-2
|
| 23 |
+
RUN pip3 install -r requirements.txt
|
| 24 |
+
RUN python3 download_model.py 124M
|
| 25 |
+
RUN python3 download_model.py 355M
|
| 26 |
+
RUN python3 download_model.py 774M
|
| 27 |
+
RUN python3 download_model.py 1558M
|