diff --git a/setup.py b/setup.py index 6eaadb37..38998ef4 100644 --- a/setup.py +++ b/setup.py @@ -16,7 +16,7 @@ ] extras_require = { - "cuda": ["cupy >= 9.0.0", "jinja2 >= 2.10"], + "cuda": ["cupy >= 9.0.0", "jinja2 >= 2.10", "setuptools"], "dask": ["dask[array] >= 2.2.0"], "jax": ["jax >= 0.2.11", "jaxlib >= 0.1.65"], "scipy": ["scipy >= 1.4.0"],