Commit
•
116c902
1
Parent(s):
b868983
Use 23.06 version
Browse files
Makefile
CHANGED
|
@@ -1,11 +1,10 @@
|
|
| 1 |
-
ONNX_PATH=dinov2_vitl14/1
|
| 2 |
MODEL_PATH=model_repository/dinov2_vitl14/1
|
| 3 |
|
| 4 |
|
| 5 |
model:
|
| 6 |
git lfs install
|
| 7 |
git clone git@hf.co:RoundtTble/dinov2_vitl14_onnx
|
| 8 |
-
cp -r dinov2_vitl14_onnx/model_repository/dinov2_vitl14/ .
|
| 9 |
rm -rf dinov2_vitl14_onnx
|
| 10 |
|
| 11 |
|
|
@@ -14,13 +13,12 @@ trt:
|
|
| 14 |
docker run --gpus all -it \
|
| 15 |
-v $(CURDIR)/$(ONNX_PATH):/workspace/onnx \
|
| 16 |
-v $(CURDIR)/$(MODEL_PATH):/workspace/$(MODEL_PATH) \
|
| 17 |
-
--rm nvcr.io/nvidia/pytorch:23.
|
| 18 |
/bin/bash -c \
|
| 19 |
"trtexec --onnx=onnx/model.onnx --saveEngine=$(MODEL_PATH)/model.plan \
|
| 20 |
--minShapes=input:1x3x560x560 \
|
| 21 |
--optShapes=input:2x3x560x560 \
|
| 22 |
-
--maxShapes=input:8x3x560x560
|
| 23 |
-
--fp16"
|
| 24 |
|
| 25 |
|
| 26 |
triton:
|
|
|
|
| 1 |
+
ONNX_PATH=dinov2_vitl14_onnx/model_repository/dinov2_vitl14/1
|
| 2 |
MODEL_PATH=model_repository/dinov2_vitl14/1
|
| 3 |
|
| 4 |
|
| 5 |
model:
|
| 6 |
git lfs install
|
| 7 |
git clone git@hf.co:RoundtTble/dinov2_vitl14_onnx
|
|
|
|
| 8 |
rm -rf dinov2_vitl14_onnx
|
| 9 |
|
| 10 |
|
|
|
|
| 13 |
docker run --gpus all -it \
|
| 14 |
-v $(CURDIR)/$(ONNX_PATH):/workspace/onnx \
|
| 15 |
-v $(CURDIR)/$(MODEL_PATH):/workspace/$(MODEL_PATH) \
|
| 16 |
+
--rm nvcr.io/nvidia/pytorch:23.06-py3 \
|
| 17 |
/bin/bash -c \
|
| 18 |
"trtexec --onnx=onnx/model.onnx --saveEngine=$(MODEL_PATH)/model.plan \
|
| 19 |
--minShapes=input:1x3x560x560 \
|
| 20 |
--optShapes=input:2x3x560x560 \
|
| 21 |
+
--maxShapes=input:8x3x560x560"
|
|
|
|
| 22 |
|
| 23 |
|
| 24 |
triton:
|