# FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
FROM nvidia/cudagl:9.0-devel-ubuntu16.04

ARG USER_NAME
ARG USER_PASSWORD
ARG USER_ID
ARG USER_GID

RUN apt-get update
RUN apt install sudo
RUN useradd -ms /bin/bash $USER_NAME
RUN usermod -aG sudo $USER_NAME
RUN yes $USER_PASSWORD | passwd $USER_NAME

# set uid and gid to match those outside the container
RUN usermod -u $USER_ID $USER_NAME
RUN groupmod -g $USER_GID $USER_NAME

# work directory
WORKDIR /home/$USER_NAME

# FIX for cudnn7 which is not in the base image
ENV NVIDIA_CUDNN_VERSION=7.1.3.16-1+cuda9.0
RUN apt-get update && apt-get install -y --no-install-recommends \
libcudnn7=${NVIDIA_CUDNN_VERSION} \
libcudnn7-dev=${NVIDIA_CUDNN_VERSION} \
&& apt-mark hold libcudnn7 \
&& rm -rf /var/lib/apt/lists/*

# install system dependencies
COPY ./scripts/install_deps.sh /tmp/install_deps.sh
RUN yes "Y" | /tmp/install_deps.sh

# setup python environment
RUN cd $WORKDIR
ENV VIRTUAL_ENV=/home/$USER_NAME/alfred_env
RUN python3 -m virtualenv --python=/usr/bin/python3 $VIRTUAL_ENV
ENV PATH="$VIRTUAL_ENV/bin:$PATH"

# install python requirements
RUN pip install --upgrade pip==20.3.4
RUN pip install -U setuptools
COPY ./requirements.txt /tmp/requirements.txt
RUN pip install -r /tmp/requirements.txt

# install GLX-Gears (for debugging)
RUN apt-get update && apt-get install -y \
   mesa-utils && \
   rm -rf /var/lib/apt/lists/*

# change ownership of everything to our user
RUN mkdir /home/$USER_NAME/alfred
RUN cd ${USER_HOME_DIR} && echo $(pwd) && chown $USER_NAME:$USER_NAME -R .

# copy scripts
COPY ./scripts/startx.py /home/$USER_NAME/

ENTRYPOINT bash -c "export ALFRED_ROOT=~/alfred && /bin/bash"