Perth0603 commited on
Commit
2372a04
·
verified ·
1 Parent(s): 00007cf

Upload requirements.txt with huggingface_hub

Browse files
Files changed (1) hide show
  1. 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
+