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