Skip to content
Permalink
Browse files

update dockerfile

  • Loading branch information...
ding
ding committed Apr 6, 2019
1 parent 4f24df6 commit d584248a57303a16f9f37aacd019dd58e4582e38
Showing with 2 additions and 2 deletions.
  1. +2 −2 docker/Dockerfile
@@ -20,13 +20,13 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
ln -s cuda-9.0 /usr/local/cuda && \
rm -rf /var/lib/apt/lists/*

RUN apt-get update && apt-get install -y --no-install-recommends \
RUN apt-get update && apt-get install -y --allow-unauthenticated --no-install-recommends \
cuda-libraries-$CUDA_PKG_VERSION \
libnccl2=$NCCL_VERSION-1+cuda9.0 && \
apt-mark hold libnccl2 && \
rm -rf /var/lib/apt/lists/*

RUN apt-get update && apt-get install -y --no-install-recommends \
RUN apt-get update && apt-get install -y --allow-unauthenticated --no-install-recommends \
cuda-libraries-dev-$CUDA_PKG_VERSION \
cuda-nvml-dev-$CUDA_PKG_VERSION \
cuda-minimal-build-$CUDA_PKG_VERSION \

0 comments on commit d584248

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