

FROM nvcr.io/nvidia/pytorch:23.03-py3
ENV DEBIAN_FRONTEND=noninteractive
COPY requirements.txt requirements.txt
#USER guest
### Mujoco requires this
RUN apt-get update 
RUN apt-get -y install libglfw3 
RUN pip install --upgrade pip
RUN pip install -r requirements.txt
RUN pip install --upgrade "jax[cuda12_pip]" -f https://storage.googleapis.com/jax-releases/jax_cuda_releases.html









