FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-devel
#FROM getkeops/keops-full:2.1-geomloss0.2.5-cuda11.8-pytorch2.0.0-python3.10
# FROM pytorch/pytorch:2.1.0-cuda11.8-cudnn8-devel
RUN apt-get update
RUN apt-get install -y git

COPY requirements.txt /tmp/requirements.txt

RUN pip install -r /tmp/requirements.txt