mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
29 lines
985 B
Plaintext
29 lines
985 B
Plaintext
FROM ubuntu:16.04
|
|
|
|
LABEL com.nvidia.volumes.needed="nvidia_driver"
|
|
RUN apt-get update && apt-get install -y --no-install-recommends \
|
|
build-essential \
|
|
git \
|
|
curl \
|
|
ca-certificates \
|
|
libjpeg-dev \
|
|
libpng-dev && \
|
|
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 conda-build && \
|
|
/opt/conda/bin/conda install numpy pyyaml scipy ipython&& \
|
|
/opt/conda/bin/conda clean -ya
|
|
ENV PATH /opt/conda/bin:$PATH
|
|
RUN conda install pytorch torchvision cuda90 -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
|
|
RUN chmod -R a+w /workspace
|