Parlai container

Dockerfile

@@ -0,0 +1,33 @@
+# Copyright (c) Meta, Inc. and its affiliates.
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04
+
+# Installing the required packages
+RUN apt update -y
+RUN apt install -y git curl
+
+# Installing Anaconda
+WORKDIR /root
+RUN curl https://repo.anaconda.com/archive/Anaconda3-2021.05-Linux-x86_64.sh -o anaconda_installer.sh
+RUN bash anaconda_installer.sh -b -p
+ENV PATH="/root/anaconda3/bin:$PATH"
+
+# Installing recommmended pre-requirements
+RUN conda install pytorch torchvision torchaudio -c pytorch-lts -c nvidia
+RUN pip install spacy==3.2.4 tokenizers pandas transformers fairseq contractions boto3==1.17.95 botocore==1.20.95
+
+# Configuring packages for English
+RUN python -m spacy download en_core_web_sm
+RUN echo "import nltk; nltk.download(\"stopwords\"); nltk.download(\"punkt\")" > nltk_dl_script.py
+RUN python nltk_dl_script.py
+
+# Download the ParlAI Github repo
+RUN git clone https://github.com/facebookresearch/ParlAI.git ~/ParlAI
+
+# Running ParlAI install
+RUN cd ~/ParlAI && \
+    pip install -r requirements.txt && \
+    python setup.py develop
+
+CMD ["parlai", "party"]