safraeli commited on
Commit
d9ee5b0
·
verified ·
1 Parent(s): 5ef3341

Pin xgboost<3 to skip nvidia-nccl-cu12 (294MB)

Browse files
Files changed (1) hide show
  1. backend/requirements-docker.txt +1 -1
backend/requirements-docker.txt CHANGED
@@ -6,7 +6,7 @@ numpy>=1.24
6
  scikit-learn>=1.3
7
  requests>=2.28
8
  python-dotenv>=1.0
9
- xgboost>=2.0
10
  pvlib>=0.10.0
11
  astral>=3.2
12
  google-genai>=1.0
 
6
  scikit-learn>=1.3
7
  requests>=2.28
8
  python-dotenv>=1.0
9
+ xgboost>=2.0,<3.0 # pin <3 to avoid nvidia-nccl-cu12 (294MB, useless on CPU)
10
  pvlib>=0.10.0
11
  astral>=3.2
12
  google-genai>=1.0