Skip to content

M
maskrcnn
Commit 43cb539a authored by keineahnung2345's avatar keineahnung2345 Committed by Francisco Massa
Browse files
Options

add opencv support to Dockerfile (#183) 

* add opencv support to Dockerfile

cv2 is used by predictor.py, so it's required

* install opencv-python from conda
parent dea7f9ea
Show whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
docker/Dockerfile
View file @ 43cb539a
...@@ -30,8 +30,10 @@ ENV CONDA_AUTO_UPDATE_CONDA=false ...@@ -30,8 +30,10 @@ ENV CONDA_AUTO_UPDATE_CONDA=false
RUN conda install -y ipython RUN conda install -y ipython
RUN pip install ninja yacs cython matplotlib RUN pip install ninja yacs cython matplotlib
# Install PyTorch 1.0 Nightly # Install PyTorch 1.0 Nightly and OpenCV
RUN conda install -y pytorch-nightly -c pytorch && conda clean -ya RUN conda install -y pytorch-nightly -c pytorch \
&& conda install -y opencv -c menpo \
&& conda clean -ya
# Install TorchVision master # Install TorchVision master
RUN git clone https://github.com/pytorch/vision.git \ RUN git clone https://github.com/pytorch/vision.git \
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment