antitheft159 commited on
Commit
f595b5c
1 Parent(s): f150e1c

Upload palabranoelani_dockerfile_gpu_195.py

Browse files
palabranoelani_dockerfile_gpu_195.py ADDED
@@ -0,0 +1,27 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ # -*- coding: utf-8 -*-
2
+ """palabraNoelani/Dockerfile.gpu.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