OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

fix nccl version

Browse Source
This commit is contained in:
ngimel 2017-09-29 15:21:05 -07:00 committed by Soumith Chintala
parent 5cc3aff9ba
commit 60bff0a5f3
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
Dockerfile
View File

@ -9,7 +9,8 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
curl \
vim \
ca-certificates \
libnccl-dev \
libnccl2=2.0.5-2+cuda8.0 \
libnccl-dev=2.0.5-2+cuda8.0 \
libjpeg-dev \
libpng-dev &&\
rm -rf /var/lib/apt/lists/*