mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-23 00:56:35 +00:00
update requirements.txt
This commit is contained in:
+4
-1
@@ -1,7 +1,10 @@
|
||||
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
|
||||
|
||||
Reference in New Issue
Block a user