File size: 451 Bytes
dfafaa4
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
# GPU build (install torch/torchvision/torchaudio separately per https://pytorch.org/get-started/locally/)
# Example (CUDA 12.1):
#   pip install --index-url https://download.pytorch.org/whl/cu121 torch torchvision torchaudio

# Core libs (do NOT pin torch* here; users install matching CUDA builds first)
torchmetrics
torchcam
captum
lightning
albumentations
opencv-python-headless
matplotlib
seaborn
tensorboard
rich
PyYAML
streamlit
requests
gradio