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 bf82e60 commit d8fa557bc3d47138595f81747d9098fffaf5bc59
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 d8fa557

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