Create Dockerfile and add some extra libs to requirements.txt

Dockerfile

@@ -0,0 +1,29 @@
+FROM nvidia/cuda:10.2-base-ubuntu16.04
+
+RUN apt-get update \
+    && DEBIAN_FRONTEND=noninteractive apt-get install -y locales \
+    && sed -i -e 's/# en_US.UTF-8 UTF-8/en_US.UTF-8 UTF-8/' /etc/locale.gen \
+    && dpkg-reconfigure --frontend=noninteractive locales \
+    && update-locale LANG=en_US.UTF-8
+
+ENV LANG en_US.UTF-8 
+ENV LC_ALL en_US.UTF-8
+
+RUN apt-get update -y && apt-get install -y software-properties-common && add-apt-repository ppa:deadsnakes/ppa
+
+RUN apt-get update \
+ && apt-get install -y curl unzip \
+    python3.6 python3-pip python3-setuptools \
+ && ln -s /usr/bin/python3.6 /usr/bin/python \
+ && apt-get clean \
+ && rm -rf /var/lib/apt/lists/*
+
+ # update pip
+RUN python3.6 -m pip install pip --upgrade
+RUN python3.6 -m pip install wheel
+
+COPY requirements.txt /src/
+
+WORKDIR /src
+RUN pip3.6 install -r /src/requirements.txt
+

requirements.txt

@@ -0,0 +1,26 @@
+cffi==1.14.2
+cmake==3.18.0
+cycler==0.10.0
+decorator==4.4.2
+future==0.18.2
+imageio==2.9.0
+intel-openmp==2020.0.133
+kiwisolver==1.1.0
+matplotlib==3.0.3
+mkl==2019.0
+mkl-include==2019.0
+networkx==2.4
+numpy==1.18.5
+Pillow==7.2.0
+pycparser==2.20
+pyparsing==2.4.7
+python-dateutil==2.8.1
+PyWavelets==1.1.1
+PyYAML==5.3.1
+scikit-image==0.15.0
+scipy==1.4.1
+six==1.15.0
+torch==0.4.1
+torchvision==0.2.1
+tqdm==4.19.9
+typing==3.7.4.3