FROM nvidia/cuda:11.3.0-runtime-ubuntu20.04

ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Europe/Moscow

RUN mkdir "/data"
COPY data /data

RUN apt-get update --fix-missing
RUN apt-get install -y python3 python3-pip psmisc git libturbojpeg ffmpeg libsm6 libxext6 wget

ENV PATH="/root/miniconda3/bin:${PATH}"
ARG PATH="/root/miniconda3/bin:${PATH}"

RUN wget https://repo.anaconda.com/miniconda/Miniconda3-latest-Linux-x86_64.sh
RUN mkdir /root/.conda
RUN bash Miniconda3-latest-Linux-x86_64.sh -b
RUN rm -f Miniconda3-latest-Linux-x86_64.sh

COPY environment.yml .
RUN conda env create -f environment.yml

SHELL ["conda", "run", "-n", "openood", "/bin/bash", "-c"]
RUN pip install libmr==0.1.9 wandb==0.14.0
RUN pip install --upgrade --no-cache-dir --trusted-host download.pytorch.org --extra-index-url https://download.pytorch.org/whl/cu113 torch==1.12.1+cu113 torchvision==0.13.1+cu113
