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:
|