Skip to content

M
maskrcnn
Commit bd39d2c1 authored by Miguel Varela Ramos's avatar Miguel Varela Ramos Committed by Francisco Massa
Browse files
Options

Fixes conda CUDA version in Dockerfile (#604) 

* fixes to dockerfile

* replaces local installation by git clone
parent 4a7dcc4d
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 2 deletions
docker/Dockerfile
View file @ bd39d2c1
......@@ -31,8 +31,9 @@ ENV CONDA_AUTO_UPDATE_CONDA=false
RUN conda install -y ipython
RUN pip install ninja yacs cython matplotlib opencv-python
# Install PyTorch 1.0 Nightly and OpenCV
RUN conda install -y pytorch-nightly -c pytorch \
# Install PyTorch 1.0 Nightly
ARG CUDA
RUN echo conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch \
&& conda clean -ya
# Install TorchVision master
......
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