antitheft159
commited on
Commit
•
79c58f9
1
Parent(s):
f595b5c
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
|