| FROM nvidia/cuda:9.1-cudnn7-devel-ubuntu16.04 | |
| RUN apt-get update && apt-get install -y --no-install-recommends \ | |
| curl \ | |
| vim \ | |
| ca-certificates &&\ | |
| rm -rf /var/lib/apt/lists/* | |
| 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 -y python=$PYTHON_VERSION numpy pyyaml scipy ipython mkl mkl-include cython typing && \ | |
| /opt/conda/bin/conda install -y -c pytorch magma-cuda90 && \ | |
| /opt/conda/bin/conda clean -ya | |
| ENV PATH /opt/conda/bin:$PATH | |
| RUN conda install pytorch torchvision cuda91 -c pytorch | |