-
Notifications
You must be signed in to change notification settings - Fork 208
/
Dockerfile-jetson
31 lines (27 loc) · 1.78 KB
/
Dockerfile-jetson
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
# docker build -t tensorrt_inference:2.0.0 .
FROM nvcr.io/nvidia/l4t-tensorrt:r8.5.2.2-devel
WORKDIR /home/tensorrt_inference/
# apt-get 安装
RUN apt-get update
RUN apt-get install software-properties-common -y
RUN apt-get update && apt-get -y upgrade && apt-get -y install ssh vim build-essential cmake gdb git libgtk2.0-dev pkg-config \
libavcodec-dev libavformat-dev libswscale-dev python-dev python-numpy libtbb2 libtbb-dev libjpeg-dev libpng-dev libtiff-dev \
libdc1394-22-dev qtbase5-dev qtdeclarative5-dev python3-pip zip
# opencv 安装
RUN bash -xc "curl -O https://github.com/opencv/opencv/archive/4.7.0.zip && unzip opencv-4.7.0.zip && mv opencv-4.7.0 opencv && \
curl -O https://github.com/opencv/opencv_contrib/archive/4.7.0.zip && unzip opencv_contrib-4.7.0.zip && mv opencv_contrib-4.7.0 opencv_contrib && \
pushd opencv>&1 > /dev/null && mkdir build && pushd build>&1 > /dev/null && \
cmake -D WITH_QT=ON \
-D WITH_CUDA=ON \
-D BUILD_TIFF=ON \
-D BUILD_TESTS=OFF \
-D BUILD_PERF_TESTS=OFF \
-D OPENCV_GENERATE_PKGCONFIG=ON \
-D CMAKE_INSTALL_PREFIX=/usr/local \
-D OPENCV_EXTRA_MODULES_PATH=../../opencv_contrib/modules/ \
-D BUILD_opencv_xfeatures2d=OFF .. && \
make -j4 && make -j4 install && pkg-config --cflags opencv4 && echo '/usr/local/lib' > /etc/ld.so.conf.d/opencv.conf && \
popd 2>&1 > /dev/null && popd 2>&1 > /dev/null && rm -rf opencv-4.7.0.zip && rm -rf opencv_contrib-4.7.0.zip"
# # pip3 安装
# RUN python3 -m pip install -U pip -i https://pypi.tuna.tsinghua.edu.cn/simple && pip3 install torch torchvision mxnet-cu102 onnx-simplifier -i https://pypi.tuna.tsinghua.edu.cn/simple \
# && pip3 install --ignore-installed -U PyYAML -i https://pypi.tuna.tsinghua.edu.cn/simple