From 311d6cc3e493bb68ed365cba6479d3f383cd6016 Mon Sep 17 00:00:00 2001
From: Miguel Varela Ramos <miguelvramos@ist.utl.pt>
Date: Tue, 26 Mar 2019 13:06:55 +0100
Subject: [PATCH] Fixes conda CUDA version in Dockerfile (#604)

* fixes to dockerfile

* replaces local installation by git clone
---
 docker/Dockerfile | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/docker/Dockerfile b/docker/Dockerfile
index 39b508258..58b924cf4 100644
--- 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