mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 11:26:36 +00:00
Alex Bezdieniezhnykh
6e5153ffb7
set cryptography lib to the certain version add manual_run for manual operations, right now it is onnx conversion and upload
21 lines
312 B
Plaintext
21 lines
312 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==44.0.2
|
|
numpy==1.26.4
|
|
requests
|
|
pyyaml
|
|
boto3
|
|
msgpack
|
|
rstream
|
|
onnxruntime-gpu
|
|
netron
|
|
pycuda
|
|
tensorrt |