16 lines
472 B
Plaintext
16 lines
472 B
Plaintext
# === Базовые ===
|
|
mlflow>=2.11.0
|
|
scikit-learn>=1.3.0
|
|
pandas>=2.0.0
|
|
numpy>=1.24.0
|
|
matplotlib>=3.7.0
|
|
|
|
# === Для GPU-версии (PyTorch + 3090) ===
|
|
# Устанавливайте torch отдельно под вашу CUDA:
|
|
# pip install torch --index-url https://download.pytorch.org/whl/cu121
|
|
torch>=2.1.0
|
|
torchvision>=0.16.0
|
|
|
|
# === Для MLflow UI / Model Registry ===
|
|
# SQLite backend уже входит в стандартную поставку Python
|