Skip to content
Permalink
Browse files

#288: use pytorch 1.1.0 cuda9.0.176_cudnn7.5.1_0 as found on https://…

  • Loading branch information...
Pavel Reich
Pavel Reich committed May 14, 2019
1 parent c4e74d6 commit 3fb10a4a89f979aee31bddb948aca464f904d0df
Showing with 1 addition and 1 deletion.
  1. +1 −1 docker/Dockerfile
@@ -101,7 +101,7 @@ ENV PATH=$CONDA_PREFIX/bin:$PATH
RUN conda install -y -c pytorch \
cuda90=1.0 \
magma-cuda90=2.4.0 \
"pytorch=1.0.0=py3.6_cuda9.0.176_cudnn7.4.1_1" \
"pytorch=1.1.0=py3.6_cuda9.0.176_cudnn7.5.1_0" \
torchvision=0.2.1 \
&& conda clean -ya

0 comments on commit 3fb10a4

Please sign in to comment.
You can’t perform that action at this time.