also updated dockerfile

nysbc · Aug 31, 2018 · 70b859b · 70b859b
1 parent 6e3cde7
commit 70b859b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/Dockerfile b/Dockerfile
@@ -19,7 +19,7 @@ RUN curl -o ~/miniconda.sh -O  https://repo.continuum.io/miniconda/Miniconda3-la
      /opt/conda/bin/conda clean -ya 
 
 ENV PATH /opt/conda/envs/$NAME/bin:$PATH
-RUN conda install --name $NAME -c soumith pytorch torchvision cuda80
+RUN conda install --name $NAME -c soumith pytorch=0.2.0 torchvision cuda80
 
 # setup topaz install directory
 WORKDIR /opt/topaz