adding DockerFile

37378898 · thomwolf · 4a4b0e57 · 37378898
Commit 37378898 authored Dec 14, 2018 by thomwolf
Hide whitespace changes
Inline Side-by-side

Showing with 7 additions and 0 deletions

docker/Dockerfile docker/Dockerfile +7 -0

No files found.
--- a/docker/Dockerfile
+++ b/docker/Dockerfile
+FROM pytorch/pytorch:latest
+
+RUN git clone https://github.com/NVIDIA/apex.git && cd apex && python setup.py install --cuda_ext --cpp_ext
+
+RUN pip install pytorch-pretrained-bert
+
+WORKDIR /workspace
\ No newline at end of file