diff options
| author | Natalia Gimelshein <ngimelshein@nvidia.com> | 2017-09-29 11:39:00 -0700 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2017-09-30 16:17:20 -0400 |
| commit | 5cc3aff9ba4a9a60a5b072d030bdafa786b60f66 (patch) | |
| tree | f7369aaf655300c4b6a05c1093fe72846fc8f97a /Dockerfile | |
| parent | 9b9704e7015217941b41af24fd6644f9f8aca5be (diff) | |
| download | pytorch-5cc3aff9ba4a9a60a5b072d030bdafa786b60f66.tar.gz pytorch-5cc3aff9ba4a9a60a5b072d030bdafa786b60f66.tar.bz2 pytorch-5cc3aff9ba4a9a60a5b072d030bdafa786b60f66.zip | |
use nccl deb in Dockerfile, easier to change python version
Diffstat (limited to 'Dockerfile')
| -rw-r--r-- | Dockerfile | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/Dockerfile b/Dockerfile index ad9a4b2d13..33c2ffcdca 100644 --- a/Dockerfile +++ b/Dockerfile @@ -9,19 +9,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ curl \ vim \ ca-certificates \ + libnccl-dev \ libjpeg-dev \ libpng-dev &&\ rm -rf /var/lib/apt/lists/* -RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-4.2.12-Linux-x86_64.sh && \ + +ENV PYTHON_VERSION=3.6 +RUN curl -o ~/miniconda.sh -O https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh && \ chmod +x ~/miniconda.sh && \ ~/miniconda.sh -b -p /opt/conda && \ rm ~/miniconda.sh && \ - /opt/conda/bin/conda install conda-build && \ - /opt/conda/bin/conda create -y --name pytorch-py35 python=3.5.2 numpy pyyaml scipy ipython mkl&& \ +# /opt/conda/bin/conda install conda-build && \ + /opt/conda/bin/conda create -y --name pytorch-py$PYTHON_VERSION python=$PYTHON_VERSION numpy pyyaml scipy ipython mkl&& \ /opt/conda/bin/conda clean -ya -ENV PATH /opt/conda/envs/pytorch-py35/bin:$PATH -RUN conda install --name pytorch-py35 -c soumith magma-cuda80 +ENV PATH /opt/conda/envs/pytorch-py$PYTHON_VERSION/bin:$PATH +RUN conda install --name pytorch-py$PYTHON_VERSION -c soumith magma-cuda80 # This must be done before pip so that requirements.txt is available WORKDIR /opt/pytorch COPY . . |