make TensorRT works (and break edgetpu)

This commit is contained in:
YS
2022-01-20 14:01:57 +03:00
parent 47f3d7c460
commit 76d46fe5ed
10 changed files with 418 additions and 113 deletions
+1
View File
@@ -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 \
+2
View File
@@ -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 \