Skip to content

Commit ee17bab

Browse files
committed
adds opencv and fixes numpy version
1 parent 91eb2e8 commit ee17bab

File tree

1 file changed

+6
-3
lines changed

1 file changed

+6
-3
lines changed

docker/Dockerfile

+6-3
Original file line numberDiff line numberDiff line change
@@ -23,8 +23,9 @@ RUN pip install mmsegmentation==0.14.1
2323

2424
# Install MMDetection3D requirements
2525
RUN conda clean --all
26-
RUN git clone https://github.com/TuSimple/SST.git /sst
26+
RUN git clone https://github.com/tusen-ai/SST.git /sst
2727
WORKDIR /sst
28+
RUN git checkout 8b795c40632c3868b091dc032fff7099c0d21bc5
2829
RUN pip install llvmlite --ignore-installed
2930
RUN pip install pyyaml --ignore-installed
3031
RUN pip install -r requirements/build.txt
@@ -43,6 +44,8 @@ RUN pip install git+https://github.com/openai/CLIP.git
4344
RUN pip install \
4445
wandb \
4546
pipenv \
46-
pytorch_lightning==1.6.2
47+
pytorch_lightning==1.6.2
4748

48-
RUN pip install --upgrade numpy
49+
RUN pip install "numpy<1.24.0"
50+
51+
RUN pip install opencv-python==4.5.5.64

0 commit comments

Comments
 (0)