diff --git a/python/requirements/ml/dl-cpu-requirements.txt b/python/requirements/ml/dl-cpu-requirements.txt index ccb02c8ec1cc..23902467833a 100644 --- a/python/requirements/ml/dl-cpu-requirements.txt +++ b/python/requirements/ml/dl-cpu-requirements.txt @@ -26,4 +26,4 @@ torch-cluster==1.6.3 torch-spline-conv==1.2.2 torch-geometric==2.5.3 -cupy-cuda12x==13.1.0; sys_platform != 'darwin' +cupy-cuda12x==13.6.0; sys_platform != 'darwin' diff --git a/python/requirements/ml/dl-gpu-requirements.txt b/python/requirements/ml/dl-gpu-requirements.txt index ab46a6df8157..ed43c5d15557 100644 --- a/python/requirements/ml/dl-gpu-requirements.txt +++ b/python/requirements/ml/dl-gpu-requirements.txt @@ -15,5 +15,5 @@ torch-sparse==0.6.18+pt23cu121 torch-cluster==1.6.3+pt23cu121 torch-spline-conv==1.2.2+pt23cu121 -cupy-cuda12x==13.1.0; sys_platform != 'darwin' +cupy-cuda12x==13.6.0; sys_platform != 'darwin' nixl==0.4.0; sys_platform != 'darwin'