We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 77c7ae5 commit 137065aCopy full SHA for 137065a
.gitignore
@@ -1,2 +1,3 @@
1
__pycache__
2
models/
3
+samples
Dockerfile
@@ -0,0 +1,20 @@
+FROM tensorflow/tensorflow:1.12.0-gpu-py3
+
+# nvidia-docker 1.0
4
+LABEL com.nvidia.volumes.needed="nvidia_driver"
5
+LABEL com.nvidia.cuda.version="${CUDA_VERSION}"
6
7
+# nvidia-container-runtime
8
+ENV NVIDIA_VISIBLE_DEVICES=all \
9
+ NVIDIA_DRIVER_CAPABILITIES=compute,utility \
10
+ NVIDIA_REQUIRE_CUDA="cuda>=8.0" \
11
+ LANG=C.UTF-8
12
13
+RUN apt-get update && mkdir /gpt-2
14
+ADD requirements.txt /gpt-2/requirements.txt
15
+RUN pip3 install -r /gpt-2/requirements.txt
16
17
+WORKDIR /gpt-2
18
+ADD . /gpt-2
19
20
+CMD ["sleep", "infinity"]
0 commit comments