From 772a32f137d31ed0c8875a808e2090aa90d8ca1f Mon Sep 17 00:00:00 2001 From: Jinchao Li <38700695+jincli@users.noreply.github.com> Date: Fri, 4 Sep 2020 03:07:53 -0700 Subject: [PATCH] Add dockerfile (#98) * fix nlg template * add dockerfile --- Dockerfile | 51 +++++++++++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 51 insertions(+) create mode 100644 Dockerfile diff --git a/Dockerfile b/Dockerfile new file mode 100644 index 0000000..b5707a0 --- /dev/null +++ b/Dockerfile @@ -0,0 +1,51 @@ +FROM nvidia/cuda:10.1-cudnn7-runtime-ubuntu18.04 +ENV DEBIAN_FRONTEND noninteractive +RUN apt-get update +RUN apt-get install -y --no-install-recommends software-properties-common +RUN add-apt-repository ppa:deadsnakes/ppa +RUN apt-get update +RUN apt-get install -y --no-install-recommends python3.7 python3-pip build-essential libssl-dev libffi-dev python3.7-dev + +RUN python3.7 -m pip install --upgrade pip +RUN python3.7 -m pip install setuptools wheel + +RUN which python3.7 +RUN which pip3 + +RUN ln -f -s /usr/bin/python3.7 /usr/bin/python +RUN ln -f -s /usr/bin/pip3 /usr/bin/pip +RUN python --version + +RUN pip install nltk==3.4 +RUN pip install tqdm==4.30 +RUN pip install checksumdir==1.1 +RUN pip install dataclasses +RUN pip install visdom +RUN pip install Pillow +RUN pip install future +RUN pip install torch +RUN pip install numpy==1.15.0 +RUN pip install scipy +RUN pip install scikit-learn==0.20.3 +RUN pip install pytorch-pretrained-bert==0.6.1 +RUN pip install transformers==2.3.0 +RUN pip install tensorflow==1.14 +RUN pip install tensorboard==1.14.0 +RUN pip install tensorboardX==1.7 +RUN pip install tokenizers==0.8.0 +RUN pip install allennlp==0.9.0 +RUN pip install requests +RUN pip install simplejson +RUN pip install spacy +RUN pip install unidecode +RUN pip install jieba +RUN pip install embeddings +RUN pip install quadprog +RUN pip install pyyaml + + +RUN [ "python", "-c", "import nltk; nltk.download('stopwords')" ] + +WORKDIR /root + +CMD ["/bin/bash"] -- GitLab