docker-compose.yaml配置deepstream环境

docker-compose.yaml文件

version: '3.8'

services: 
  deepstream-dev:
    build:
        context: ..
        dockerfile: docker/Dockerfile
    image: worker:v0.0.1
    restart: always
    volumes:
        - /tmp/.X11-unix:/tmp/.X11-unix
    environment:
        - DISPLAY=$DISPLAY
    runtime: nvidia
    #ports:
    #    - "8555:8554"
    shm_size: 4g
    ulimits:
        memlock: -1
        stack: 67108864

Dockerfile文件：

FROM nvcr.io/nvidia/deepstream:5.1-21.02-triton

RUN rm /etc/apt/sources.list.d/* \
 && sed -i 's/archive.ubuntu.com/mirrors.bfsu.edu.cn/g' /etc/apt/sources.list \
 && sed -i 's/security.ubuntu.com/mirrors.bfsu.edu.cn/g' /etc/apt/sources.list

RUN ln -sf /usr/share/zoneinfo/Asia/Shanghai /etc/localtime \
 && echo 'Asia/Shanghai' >/etc/timezone


RUN apt update \
 && apt install -y language-pack-zh-hans fonts-droid-fallback fonts-wqy-zenhei fonts-wqy-microhei fonts-arphic-ukai fonts-arphic-uming\
 && apt install -y wget build-essential gcc g++ gdb make openssh-server \
 && rm -rf /var/lib/apt/lists/*

RUN sed -ri 's/^#PermitRootLogin\s+.*/PermitRootLogin yes/' /etc/ssh/sshd_config && \
 sed -ri 's/UsePAM yes/#UsePAM yes/g' /etc/ssh/sshd_config

RUN service ssh start 

EXPOSE 22

COPY cmake-3.21.0-linux-x86_64.sh /home
RUN cd /home \ 
 && chmod a+x cmake-3.21.0-linux-x86_64.sh \
 && ./cmake-3.21.0-linux-x86_64.sh --prefix=/usr/local --skip-license \
 && rm cmake-3.21.0-linux-x86_64.sh

COPY TensorRT-7.2.1.6/bin/* /workspace/TensorRT-7.2.1.6/bin/
COPY TensorRT-7.2.1.6/lib/* /workspace/TensorRT-7.2.1.6/lib/

WORKDIR /workspace

ENV LD_LIBRARY_PATH="/workspace;/workspace/TensorRT-7.2.1.6/lib/:$LD_LIBRARY_PATH"

ENTRYPOINT [ ]
CMD ["/usr/sbin/sshd","-D"]