diff --git a/Dockerfile b/Dockerfile index 572b7e27d..4ea2997fa 100644 --- a/Dockerfile +++ b/Dockerfile @@ -304,3 +304,9 @@ COPY --from=rootfs / / FROM frigate AS frigate-tensorrt RUN --mount=type=bind,from=wheels,source=/trt-wheels,target=/deps/trt-wheels \ pip3 install -U /deps/trt-wheels/*.whl + +# Dev Container w/ TRT +FROM devcontainer AS devcontainer-trt + +RUN --mount=type=bind,from=wheels,source=/trt-wheels,target=/deps/trt-wheels \ + pip3 install -U /deps/trt-wheels/*.whl diff --git a/docker-compose.yml b/docker-compose.yml index bde0ca63a..e050840ab 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -11,6 +11,7 @@ services: shm_size: "256mb" build: context: . + # Use target devcontainer-trt for TensorRT dev target: devcontainer deploy: resources: @@ -28,6 +29,8 @@ services: - /etc/localtime:/etc/localtime:ro - ./config/config.yml:/config/config.yml:ro - ./debug:/media/frigate + # Create the trt-models folder using the documented method of generating TRT models + # - ./debug/trt-models:/trt-models - /dev/bus/usb:/dev/bus/usb mqtt: container_name: mqtt