mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:26:36 +00:00
26eb3b991d
revise onnx export process fixes
21 lines
304 B
Plaintext
21 lines
304 B
Plaintext
huggingface_hub
|
|
ultralytics
|
|
--extra-index-url https://download.pytorch.org/whl/cu121
|
|
torch==2.3.0+cu121
|
|
torchvision==0.18.0+cu121
|
|
torchaudio==2.3.0+cu121
|
|
albumentations
|
|
opencv-python
|
|
matplotlib
|
|
PyYAML
|
|
cryptography
|
|
numpy==1.26.4
|
|
requests
|
|
pyyaml
|
|
boto3
|
|
msgpack
|
|
rstream
|
|
onnxruntime-gpu
|
|
netron
|
|
pycuda
|
|
tensorrt |