foundationpose_cpp/docker/foundationpose_nvidia_gpu_trt10.dockerfile at f5d4cc3d5817c006f81fcd822cad05bb87870944 · zz990099/foundationpose_cpp · GitHub

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

# Base image starts with CUDA
ARG BASE_IMG=nvcr.io/nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04
FROM ${BASE_IMG} as base
ENV BASE_IMG=nvidia/cuda:12.3.2-cudnn9-devel-ubuntu22.04

ENV TENSORRT_VERSION=10.7.0
ENV TENSORRT_PACAKGE_VERSION=10.7.0.23-1+cuda12.6

ENV DEBIAN_FRONTEND=noninteractive

RUN rm /etc/apt/sources.list && \
    echo "deb https://mirrors.ustc.edu.cn/ubuntu/ jammy main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb https://mirrors.ustc.edu.cn/ubuntu/ jammy-security main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb https://mirrors.ustc.edu.cn/ubuntu/ jammy-updates main restricted universe multiverse" >> /etc/apt/sources.list && \
    echo "deb https://mirrors.ustc.edu.cn/ubuntu/ jammy-backports main restricted universe multiverse" >> /etc/apt/sources.list && \
    apt-get update


# Install basic dependencies
RUN apt install -y \
    build-essential \
    manpages-dev \
    wget \
    zlib1g \
    software-properties-common \
    git \
    libssl-dev \
    zlib1g-dev \
    libbz2-dev \
    libreadline-dev \
    libsqlite3-dev \
    wget \
    ca-certificates \
    curl \
    llvm \
    libncurses5-dev \
    xz-utils tk-dev \
    libxml2-dev \
    libxmlsec1-dev \
    libffi-dev \
    liblzma-dev \
    mecab-ipadic-utf8 \
    cmake \
    libopencv-dev \
    libeigen3-dev \
    libgoogle-glog-dev \
    libgtest-dev

# Install TensorRT + dependencies
RUN apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/3bf863cc.pub
RUN add-apt-repository "deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/ /"
RUN apt-get update
RUN TENSORRT_MAJOR_VERSION=`echo ${TENSORRT_VERSION} | cut -d '.' -f 1` && \
    apt-get install -y libnvinfer${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-plugin${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-headers-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-headers-plugin-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-plugin-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvonnxparsers${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvonnxparsers-dev=${TENSORRT_PACAKGE_VERSION} \
                      #  libnvparsers${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                      #  libnvparsers-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-lean${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-lean-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-dispatch${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-dispatch-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-vc-plugin${TENSORRT_MAJOR_VERSION}=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-vc-plugin-dev=${TENSORRT_PACAKGE_VERSION} \
                       libnvinfer-samples=${TENSORRT_PACAKGE_VERSION}

RUN cd /usr/src/tensorrt/samples \
    && make -j

# foundationpose dependencies
RUN apt-get install libassimp-dev -y

RUN cd /tmp && \
    wget https://gp.zz990099.cn/https://github.com/CVCUDA/CV-CUDA/releases/download/v0.12.0-beta/cvcuda-lib-0.12.0_beta-cuda12-x86_64-linux.deb && \
    wget https://gp.zz990099.cn/https://github.com/CVCUDA/CV-CUDA/releases/download/v0.12.0-beta/cvcuda-dev-0.12.0_beta-cuda12-x86_64-linux.deb && \
    dpkg -i cvcuda-lib-0.12.0_beta-cuda12-x86_64-linux.deb && \
    dpkg -i cvcuda-dev-0.12.0_beta-cuda12-x86_64-linux.deb && \
    rm cvcuda-lib-0.12.0_beta-cuda12-x86_64-linux.deb && \
    rm cvcuda-dev-0.12.0_beta-cuda12-x86_64-linux.deb