Dockerfile

FROM nvidia/cuda:11.1.1-cudnn8-devel-ubuntu18.04
ENV DEBIAN_FRONTEND=noninteractive
RUN apt-get update && apt-get install git sudo -y

RUN apt-get install -y software-properties-common
RUN add-apt-repository -y ppa:deadsnakes/ppa
RUN apt-get update && apt-get install git sudo -y

RUN apt-get install --no-install-recommends -y python3.9 python3.9-dev python3-pip python3-dev python-dev python-setuptools python3-setuptools
RUN apt-get install -y python3.9-distutils
RUN apt-get install -y gfortran libopenblas-dev liblapack-dev
RUN apt-get install -y gcc g++

RUN update-alternatives --install /usr/bin/pip pip /usr/bin/pip3 1
RUN update-alternatives --install /usr/bin/python python /usr/bin/python3.9 1

RUN python -m pip install --upgrade wheel setuptools pip distlib
RUN sudo pip install --default-timeout=100 future

RUN mkdir -p /ehnn
RUN git clone --recursive https://github.com/jw9730/ehnn.git /ehnn
RUN git config --global --add safe.directory /ehnn
WORKDIR /ehnn

RUN bash install.sh
CMD ["/bin/bash"]