diff options
| author | ngimel <ngimelshein@nvidia.com> | 2018-01-18 19:07:25 -0800 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-01-18 22:07:25 -0500 |
| commit | 92aeca1279265d24493dc6ced7dde9a368faf048 (patch) | |
| tree | c30fc02519bc8c58230f6ca1296c9566828dd8b3 /tools | |
| parent | f9fd82d8933639a8cf20a029c7fa47fff8fdb93d (diff) | |
| download | pytorch-92aeca1279265d24493dc6ced7dde9a368faf048.tar.gz pytorch-92aeca1279265d24493dc6ced7dde9a368faf048.tar.bz2 pytorch-92aeca1279265d24493dc6ced7dde9a368faf048.zip | |
update runtime dockerfile (#4736)
Diffstat (limited to 'tools')
| -rw-r--r-- | tools/docker/Dockerfile_runtime | 13 |
1 files changed, 8 insertions, 5 deletions
diff --git a/tools/docker/Dockerfile_runtime b/tools/docker/Dockerfile_runtime index 57cc8dd6f9..914903a37e 100644 --- a/tools/docker/Dockerfile_runtime +++ b/tools/docker/Dockerfile_runtime @@ -10,17 +10,20 @@ RUN apt-get update && apt-get install -y --no-install-recommends \ 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 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 && /opt/conda/bin/conda clean -ya -RUN conda install --name pytorch-py35 pytorch torchvision cuda80 -c soumith && /opt/conda/bin/conda clean -ya +ENV PATH opt/conda/bin:/opt/conda/envs/pytorch-py$PYTHON_VERSION/bin:$PATH +RUN conda install --name pytorch-py$PYTHON_VERSION -c pytorch magma-cuda80 && /opt/conda/bin/conda clean -ya +RUN conda install --name pytorch-py$PYTHON_VERSION pytorch torchvision cuda80 -c pytorch && /opt/conda/bin/conda clean -ya +ENV NVIDIA_VISIBLE_DEVICES all +ENV NVIDIA_DRIVER_CAPABILITIES compute,utility ENV LD_LIBRARY_PATH /usr/local/nvidia/lib:/usr/local/nvidia/lib64 WORKDIR /workspace |