FROM pytorch/pytorch:1.13.1-cuda11.6-cudnn8-runtime
ARG DEBIAN_FRONTEND=noninteractive
ENV TZ=Asia/Shanghai
# RUN rm /etc/apt/sources.list.d/cuda.list
RUN apt-get update
# 安装ifconfig命令
RUN apt-get install -y net-tools
# 安装ping命令
RUN apt-get install -y iputils-ping
# opencv-python依赖
RUN apt-get install -y libgl1 ffmpeg
RUN apt-get install -y wget
RUN pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple --upgrade pip
RUN pip3 config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple
# 安装yolov5s
RUN pip3 install yolov5==7.0.7
# 拷贝代码至docker
WORKDIR /root/
COPY envs ./envs
COPY code ./code
WORKDIR /root/envs
# 安装rocketmq-client-cpp
RUN wget https://ghproxy.com/https://github.com/apache/rocketmq-client-cpp/releases/download/2.0.0/rocketmq-client-cpp-2.0.0.amd64.deb
RUN dpkg -i rocketmq-client-cpp-2.0.0.amd64.deb
RUN pip3 install rocketmq-client-python==2.0.0
RUN pip3 install -r requirements.txt
WORKDIR /root/code
RUN wget https://ghproxy.com/https://github.com/aler9/rtsp-simple-server/releases/download/v0.21.4/rtsp-simple-server_v0.21.4_linux_amd64.tar.gz
RUN tar -xzvf rtsp-simple-server_v0.21.4_linux_amd64.tar.gz
ENV OPENCV_FFMPEG_CAPTURE_OPTIONS="rtsp_transport;udp"
ENTRYPOINT ["sh", "start.sh"]