frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

CUDA 9.0 and cuDNN 7 (#5186)

Browse Source
This commit is contained in:
Amir Arsalan Soltani
2018-02-12 14:21:56 -05:00
committed by Soumith Chintala
parent 07be53b57f
commit 19c2ad8834
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
Dockerfile
View File

@ -1,4 +1,4 @@
FROM nvidia/cuda:8.0-cudnn6-devel-ubuntu16.04 FROM nvidia/cuda:9.0-cudnn7-devel-ubuntu16.04
RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list RUN echo "deb http://developer.download.nvidia.com/compute/machine-learning/repos/ubuntu1604/x86_64 /" > /etc/apt/sources.list.d/nvidia-ml.list
@ -9,8 +9,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
curl \ curl \
vim \ vim \
ca-certificates \ ca-certificates \
libnccl2=2.0.5-2+cuda8.0 \ libnccl2=2.0.5-2+cuda9.0 \
libnccl-dev=2.0.5-2+cuda8.0 \ libnccl-dev=2.0.5-2+cuda9.0 \
libjpeg-dev \ libjpeg-dev \
libpng-dev &&\ libpng-dev &&\
rm -rf /var/lib/apt/lists/* rm -rf /var/lib/apt/lists/*