diff options
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 . . |