diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt index f500bc598..d1eb4c1b2 100644 --- a/dependencies/requirements/base_requirements/requirements.txt +++ b/dependencies/requirements/base_requirements/requirements.txt @@ -19,6 +19,7 @@ jaxlib jaxopt libtpu>=0.0.42.1 Jinja2 +nvidia-cuda-cccl>=13.1.115 ; sys_platform != 'darwin' opencv-python-headless optax orbax-checkpoint diff --git a/dependencies/requirements/generated_requirements/requirements.txt b/dependencies/requirements/generated_requirements/requirements.txt index 3aee20155..66b169d12 100644 --- a/dependencies/requirements/generated_requirements/requirements.txt +++ b/dependencies/requirements/generated_requirements/requirements.txt @@ -108,7 +108,7 @@ namex>=0.1.0 networkx>=3.6.1 numpy-typing-compat>=20251206.2.0 numpy>=2.0.2 -nvidia-cuda-cccl>=13.1.115 +nvidia-cuda-cccl>=13.1.115 ; sys_platform != 'darwin' oauthlib>=3.3.1 opencv-python-headless>=4.13.0.92 opt-einsum>=3.4.0