forked from cvlab-epfl/tf-lift
-
Notifications
You must be signed in to change notification settings - Fork 0
/
dockerfile
89 lines (83 loc) · 2.88 KB
/
dockerfile
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
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
# docker build . -t tflift
# function dockerpythontflift() {
# xhost +;
# docker run --runtime=nvidia -e NVIDIA_VISIBLE_DEVICES=1 --rm -it \
# --env HOST_PERMS="$(id -u):$(id -g)" \
# --user=$(id -u) \
# --env="DISPLAY" \
# --env QT_X11_NO_MITSHM=1 \
# --workdir=/app \
# --volume="$PWD":/app \
# --volume="/home/tsael/":/home/tsael \
# --volume="/etc/group:/etc/group:ro" \
# --volume="/etc/passwd:/etc/passwd:ro" \
# --volume="/etc/shadow:/etc/shadow:ro" \
# --volume="/etc/sudoers.d:/etc/sudoers.d:ro" \
# --volume="/tmp/.X11-unix:/tmp/.X11-unix:rw" \
# --volume="/dev/input:/dev/input" \
# tflift $*;
# xhost -;}
# alias pylift="dockerpythontflift"
FROM tensorflow/tensorflow:1.4.1-gpu-py3
ENV NVIDIA_DRIVER_CAPABILITIES ${NVIDIA_DRIVER_CAPABILITIES},display
SHELL ["/bin/bash", "-c"]
RUN apt-get update && pip install numpy pillow tqdm imutils &&\
DEBIAN_FRONTEND="noninteractive" apt-get install python3-tk -y && ln -fs /usr/share/zoneinfo/America/Chicago /etc/localtime && dpkg-reconfigure --frontend noninteractive tzdata\
&& apt-get install -y \
build-essential \
cmake \
git \
wget \
unzip \
yasm \
pkg-config \
libswscale-dev \
libtbb2 \
libtbb-dev \
libjpeg-dev \
libpng-dev \
libtiff-dev \
libavformat-dev \
libpq-dev \
libgtk2.0-dev \
&& rm -rf /var/lib/apt/lists/*
WORKDIR /
ENV OPENCV_VERSION="4.1.0"
RUN wget -O opencv.zip https://github.com/opencv/opencv/archive/${OPENCV_VERSION}.zip \
&& wget -O opencv_contrib.zip https://github.com/opencv/opencv_contrib/archive/${OPENCV_VERSION}.zip \
&& unzip opencv.zip \
&& unzip opencv_contrib.zip \
&& mv opencv-${OPENCV_VERSION} opencv \
&& mv opencv_contrib-${OPENCV_VERSION} opencv_contrib \
&& mkdir /opencv/build \
&& cd /opencv/build \
# && mkdir /opencv-${OPENCV_VERSION}/cmake_binary \
# && cd /opencv-${OPENCV_VERSION}/cmake_binary \
&& cmake -DBUILD_TIFF=ON \
-DBUILD_opencv_java=OFF \
-DWITH_CUDA=OFF \
-DWITH_OPENGL=ON \
-DWITH_OPENCL=ON \
-DWITH_IPP=ON \
-DWITH_TBB=ON \
-DWITH_EIGEN=ON \
-DWITH_V4L=ON \
-DBUILD_TESTS=OFF \
-DBUILD_PERF_TESTS=OFF \
-DOPENCV_ENABLE_NONFREE=ON \
-DOPENCV_EXTRA_MODULES_PATH=/opencv_contrib/modules \
-DCMAKE_BUILD_TYPE=RELEASE \
-DCMAKE_INSTALL_PREFIX=$(python3.5 -c "import sys; print(sys.prefix)") \
-DPYTHON_EXECUTABLE=$(which python3) \
-DPYTHON_INCLUDE_DIR=$(python3.5 -c "from distutils.sysconfig import get_python_inc; print(get_python_inc())") \
-DPYTHON_PACKAGES_PATH=$(python3.5 -c "from distutils.sysconfig import get_python_lib; print(get_python_lib())") \
.. \
&& make install -j4 \
&& rm /opencv.zip \
&& rm /opencv_contrib.zip \
&& rm -r /opencv \
&& rm -r /opencv_contrib
RUN ln -s \
/usr/local/python/cv2/python-3.5/cv2.cpython-35m-x86_64-linux-gnu.so \
/usr/local/lib/python3.5/dist-packages/cv2.so
ENTRYPOINT ["/bin/bash"]