diff --git a/setup.py b/setup.py index 5c14c3ccc..f6e41d9d1 100644 --- a/setup.py +++ b/setup.py @@ -66,10 +66,10 @@ ], extras_require={ "cuda12x": [ - "cudf-cu12>=24.2", - "dask-cudf-cu12>=24.2", - "cugraph-cu12>=24.2", - "dask-cuda>=24.2", + "cudf-cu12==24.4.*", + "dask-cudf-cu12==24.4.*", + "cugraph-cu12==24.4.*", + "dask-cuda==24.4.*", "spacy[cuda12x]>=3.6.0, <4.0.0", ] },