File size: 251 Bytes
308345c
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
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