FROM qinsheng/cudot:edge

# install ubuntu dependencies
ENV DEBIAN_FRONTEND=noninteractive 
RUN sudo apt-get update && \
    sudo apt-get -y install python3-pip xvfb ffmpeg git build-essential python-opengl

# install mujoco dependencies
RUN sudo apt-get -y install wget unzip software-properties-common \
    libgl1-mesa-dev \
    libgl1-mesa-glx \
    libglew-dev \
    libosmesa6-dev patchelf

RUN sudo apt install -y libosmesa6-dev patchelf && \
    pip3 install gym && \
    pip3 install mujoco-py  && \
    mkdir -p $HOME/.mujoco/ && \
    wget https://mujoco.org/download/mujoco210-linux-x86_64.tar.gz -O mujoco.tar.gz && \
    tar -xf mujoco.tar.gz -C $HOME/.mujoco/ && \
    /bin/rm mujoco.tar.gz
    # /bin/rm mujoco.tar.gz && \
    # wget --directory-prefix $HOME/.mujoco/ https://www.roboti.us/file/mjkey.txt
RUN sudo apt install -y ssh
RUN sudo ./cleanup
ENV LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/home/USER/.mujoco/mujoco210/bin

# install python dependencies
RUN python -c "import mujoco_py"

RUN pip install torch==1.12.1+cu113 -f https://download.pytorch.org/whl/torch_stable.html