mirror of
https://github.com/blakeblackshear/frigate.git
synced 2026-07-21 03:09:02 +03:00
make TensorRT works (and break edgetpu)
This commit is contained in:
@@ -2,6 +2,7 @@
|
||||
|
||||
set -xe
|
||||
cd /tensorrt_demos/plugins && make
|
||||
cp /tensorrt_demos/plugins/libyolo_layer.so /plugin/libyolo_layer.so
|
||||
|
||||
cd /tensorrt_demos/yolo
|
||||
for model in yolov4-tiny-288 \
|
||||
|
||||
@@ -1,12 +1,14 @@
|
||||
#!/bin/bash
|
||||
|
||||
mkdir -p $(pwd)/model
|
||||
mkdir -p $(pwd)/plugin
|
||||
|
||||
docker build --tag models.yolo4 --file ./Dockerfile.l4t.tf15 ./assets/
|
||||
|
||||
sudo docker run --rm -it --name models.yolo4 \
|
||||
--mount type=tmpfs,target=/tmp/cache,tmpfs-size=1000000000 \
|
||||
-v $(pwd)/model:/model:rw \
|
||||
-v $(pwd)/plugin:/plugin:rw \
|
||||
-v /tmp/argus_socket:/tmp/argus_socket \
|
||||
-e NVIDIA_VISIBLE_DEVICES=all \
|
||||
-e NVIDIA_DRIVER_CAPABILITIES=compute,utility,video \
|
||||
|
||||
Reference in New Issue
Block a user