Skip to content

Commit 19327c1

Browse files
committed
Add a Dockerfile and document usage in README
1 parent 77c7ae5 commit 19327c1

File tree

2 files changed

+22
-0
lines changed

2 files changed

+22
-0
lines changed

.gitignore

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,2 +1,4 @@
11
__pycache__
22
models/
3+
samples
4+
core

Dockerfile

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,20 @@
1+
FROM tensorflow/tensorflow:1.12.0-gpu-py3
2+
3+
# 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

Comments
 (0)