azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 07:16:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix tensor rt engine

Browse Source
This commit is contained in:
zxsanny
2025-03-28 14:50:43 +02:00
committed by Alex Bezdieniezhnykh
parent 5b89a21b36
commit 06a23525a6
16 changed files with 272 additions and 94 deletions
Download Patch File Download Diff File Expand all files Collapse all files
requirements.txt
+2 -5
View File
@@ -1,10 +1,6 @@
huggingface_hub
torch
torchvision
torchaudio
ultralytics
albumentations
opencv-python
matplotlib
PyYAML
@@ -17,4 +13,5 @@ msgpack
rstream
onnxruntime-gpu
netron
pycuda
pycuda
tensorrt