diff options
| author | Sean Naren <seannaren@hotmail.com> | 2018-02-14 16:57:23 +0000 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2018-02-14 11:57:23 -0500 |
| commit | fe810edc805f5a377bdf71e71ac5eab2bb3d3566 (patch) | |
| tree | 79a319d80e3faa6a6878ba2ccbf2986b3ecf4f3d /tools/docker | |
| parent | 8910dd5a81cdddb2612ac1c9b93a202fcf3a6e60 (diff) | |
| download | pytorch-fe810edc805f5a377bdf71e71ac5eab2bb3d3566.tar.gz pytorch-fe810edc805f5a377bdf71e71ac5eab2bb3d3566.tar.bz2 pytorch-fe810edc805f5a377bdf71e71ac5eab2bb3d3566.zip | |
Consolidated dockerfile changes, updated README (#5235)
Diffstat (limited to 'tools/docker')
| -rw-r--r-- | tools/docker/Dockerfile9 | 39 |
1 files changed, 0 insertions, 39 deletions
diff --git a/tools/docker/Dockerfile9 b/tools/docker/Dockerfile9 deleted file mode 100644 index a6b85d22ea..0000000000 --- a/tools/docker/Dockerfile9 +++ /dev/null @@ -1,39 +0,0 @@ -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 apt-get update && apt-get install -y --no-install-recommends \ - build-essential \ - cmake \ - git \ - curl \ - vim \ - ca-certificates \ - libjpeg-dev \ - libpng-dev &&\ - rm -rf /var/lib/apt/lists/* - - -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-py$PYTHON_VERSION python=$PYTHON_VERSION numpy pyyaml scipy ipython mkl&& \ - /opt/conda/bin/conda clean -ya -ENV PATH /opt/conda/envs/pytorch-py$PYTHON_VERSION/bin:$PATH -RUN conda install --name pytorch-py$PYTHON_VERSION -c soumith magma-cuda90 -# This must be done before pip so that requirements.txt is available -WORKDIR /opt/pytorch -COPY . . - -RUN git submodule update --init -RUN TORCH_CUDA_ARCH_LIST="5.2 6.0 6.1 7.0+PTX" TORCH_NVCC_FLAGS="-Xfatbin -compress-all" \ - CMAKE_PREFIX_PATH="$(dirname $(which conda))/../" \ - pip install -v . - -RUN git clone https://github.com/pytorch/vision.git && cd vision && pip install -v . - -WORKDIR /workspace -RUN chmod -R a+w /workspace |