Upload requirements.txt with huggingface_hub
Browse files- requirements.txt +34 -0
requirements.txt
ADDED
|
@@ -0,0 +1,34 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# Core numerical and IO
|
| 2 |
+
numpy<2.0
|
| 3 |
+
pandas==2.2.2
|
| 4 |
+
joblib>=1.3
|
| 5 |
+
|
| 6 |
+
# CPU ML fallback
|
| 7 |
+
scikit-learn>=1.3
|
| 8 |
+
|
| 9 |
+
# GPU array library (imported as `cupy` in the notebook)
|
| 10 |
+
# Pick the CUDA 12.x wheel that matches your GPU drivers
|
| 11 |
+
cupy-cuda12x>=12.0.0
|
| 12 |
+
|
| 13 |
+
# Optional: used only for the simple GPU availability check in the notebook
|
| 14 |
+
# (You can skip installing torch if you remove the torch check cell.)
|
| 15 |
+
torch>=2.1
|
| 16 |
+
|
| 17 |
+
# IMPORTANT: RAPIDS (cuDF, cuML) install instructions (NOT via pip)
|
| 18 |
+
# The notebook uses RAPIDS cuDF/cuML for GPU RandomForest. Install via Conda:
|
| 19 |
+
#
|
| 20 |
+
# conda create -n rapids-24.08 -c rapidsai -c conda-forge -c nvidia \
|
| 21 |
+
# rapids=24.08 python=3.10 cuda-version=12.2 -y
|
| 22 |
+
# conda activate rapids-24.08
|
| 23 |
+
#
|
| 24 |
+
# This will install `cudf` and `cuml` compatible with CUDA 12.2. If you're on Windows,
|
| 25 |
+
# use WSL2 (Ubuntu) for best support. Native Windows support for RAPIDS is limited.
|
| 26 |
+
|
| 27 |
+
# Windows-friendly GPU fallback (no RAPIDS required)
|
| 28 |
+
# XGBoost supports GPU acceleration on native Windows when built with CUDA.
|
| 29 |
+
# pip install xgboost will fetch a prebuilt wheel with GPU support if available.
|
| 30 |
+
xgboost>=2.0
|
| 31 |
+
|
| 32 |
+
# (Optional) Another alternative with some GPU support (requires separate setup):
|
| 33 |
+
# lightgbm
|
| 34 |
+
|