Fixes conda CUDA version in Dockerfile (#604)

* fixes to dockerfile * replaces local installation by git clone
facebookresearch · Mar 26, 2019 · bd39d2c · bd39d2c
1 parent 4a7dcc4
commit bd39d2c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/docker/Dockerfile b/docker/Dockerfile
@@ -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