diff --git a/dependencies/requirements/base_requirements/requirements.txt b/dependencies/requirements/base_requirements/requirements.txt index a446ae034..37d693211 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 0e66b4637..120826e81 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