File size: 1,725 Bytes
79cf6ef |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 |
FROM nvidia/cuda:10.0-cudnn7-devel-ubuntu16.04
SHELL ["/bin/bash", "-c"]
RUN rm /etc/apt/sources.list.d/cuda.list \
/etc/apt/sources.list.d/nvidia-ml.list && \
apt-get update && \
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
software-properties-common \
wget \
git && \
add-apt-repository -y ppa:deadsnakes/ppa && \
apt-get update && \
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
python3.6 \
python3.6-dev && \
wget -O ~/get-pip.py \
https://bootstrap.pypa.io/get-pip.py && \
python3.6 ~/get-pip.py && \
pip3 --no-cache-dir install \
numpy==1.17.4 \
PyYAML==5.1.2 \
mkl==2019.0 \
mkl-include==2019.0 \
cmake==3.15.3 \
cffi==1.13.2 \
typing==3.7.4.1 \
six==1.13.0 \
Pillow==6.2.1 \
scipy==1.4.1 && \
cd /tmp && \
git clone https://github.com/pytorch/pytorch.git && \
cd pytorch && \
git checkout v1.3.0 && \
git submodule update --init --recursive && \
python3.6 setup.py install && \
cd /tmp && \
git clone https://github.com/pytorch/vision.git && \
cd vision && \
git checkout v0.4.1 && \
python3.6 setup.py install && \
DEBIAN_FRONTEND=noninteractive apt-get install -y --no-install-recommends \
ffmpeg && \
pip3 --no-cache-dir install \
opencv-python==4.1.2.30 \
albumentations==0.4.3 \
tqdm==4.39.0 \
timm==0.1.18 \
efficientnet-pytorch==0.6.3 \
ffmpeg-python==0.2.0 \
tensorflow==1.15.2 && \
cd / && \
apt-get clean && \
apt-get autoremove && \
rm -rf /var/lib/apt/lists/* /tmp/*
|