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
Show 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 ...@@ -31,8 +31,9 @@ ENV CONDA_AUTO_UPDATE_CONDA=false
RUN conda install -y ipython RUN conda install -y ipython
RUN pip install ninja yacs cython matplotlib opencv-python RUN pip install ninja yacs cython matplotlib opencv-python
# Install PyTorch 1.0 Nightly and OpenCV # Install PyTorch 1.0 Nightly
RUN conda install -y pytorch-nightly -c pytorch \ ARG CUDA
RUN echo conda install pytorch-nightly cudatoolkit=${CUDA} -c pytorch \
&& conda clean -ya && conda clean -ya
# Install TorchVision master # 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