FROM 597623766114.dkr.ecr.ap-southeast-1.amazonaws.com/nvidia-cuda:tensorrt-20.11-py3 ENV DEBIAN_FRONTEND=noninteractive RUN apt-get update RUN apt-get install ffmpeg libsm6 libxext6 -y COPY ./config /app/config WORKDIR /app/config/tensorrt_conversion RUN pip3 install -r /app/config/tensorrt_conversion/requirements.txt RUN python3.6 demo_darknet2onnx.py /app/config/tensorrt_conversion/input_volume/yolov4-tiny.cfg /app/config/tensorrt_conversion/input_volume/coco.names /app/config/tensorrt_conversion/input_volume/yolov4-tiny.weights /app/config/tensorrt_conversion/input_volume/512.jpg 1 RUN trtexec --onnx=/app/config/tensorrt_conversion/yolov4_1_3_416_416_static.onnx --explicitBatch --saveEngine=yolov4_tiny.trt --fp16 RUN python3.6 demo_trt.py /app/config/tensorrt_conversion/yolov4_tiny.trt /app/config/tensorrt_conversion/input_volume/512.jpg 416 416 #ENTRYPOINT ["bash", "-c", "python3.7 vas.py config/mobius.ini -i ${SOURCE_URL}"] #CMD nvidia-smi