18 lines
300 B
Plaintext
18 lines
300 B
Plaintext
# ===== Core =====
|
|
numpy
|
|
scipy
|
|
|
|
# ===== PyTorch (CUDA 12.4) =====
|
|
# Install via: pip install torch torchvision --index-url https://download.pytorch.org/whl/cu124
|
|
torch>=2.4
|
|
torchvision
|
|
|
|
# ===== Data loading =====
|
|
webdataset
|
|
opencv-python
|
|
|
|
# ===== Training / logging =====
|
|
tensorboard
|
|
tqdm
|
|
matplotlib
|