FROM nvcr.io/nvidia/pytorch:23.02-py3

WORKDIR /workdir

COPY requirements.txt /workdir/

# since installing icetk will install protobuf 3.18.3, and we need protobuf==3.20.3
RUN pip install -r requirements.txt && \
    pip install protobuf==3.20.3