Fixes conda CUDA version in Dockerfile (facebookresearch#604)

* fixes to dockerfile * replaces local installation by git clone
Lyears · Mar 26, 2019 · 311d6cc · 311d6cc
1 parent 1db3cc6
commit 311d6cc
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