| 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 | |
| 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 | |