diff --git a/pixi.lock b/pixi.lock index b886982c..9441c98b 100644 --- a/pixi.lock +++ b/pixi.lock @@ -5232,7 +5232,7 @@ packages: - pypi: . name: array-api-extra version: 0.7.2.dev0 - sha256: be7523c2bc2519d23eb7e92abc0466bbb70d37728f0b677e5acfd67350a2e212 + sha256: f7533831000beefe0eb2bf6b4dd47c15f7fff06c27785606f91f000cda7b0a5d requires_dist: - array-api-compat>=1.11.2,<2 requires_python: '>=3.10' diff --git a/pyproject.toml b/pyproject.toml index 507f38c9..d841294b 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -164,7 +164,7 @@ jax = "*" system-requirements = { cuda = "12" } [tool.pixi.feature.cuda-backends.target.linux-64.dependencies] -cupy = "*" +cupy = ">=6.7.0" jaxlib = { version = "*", build = "cuda12*" } pytorch = { version = "*", build = "cuda12*" } @@ -179,7 +179,7 @@ pytorch = { version = "*", build = "cuda12*" } # pytorch = { version = "*", build = "cuda12*" } # unavailable [tool.pixi.feature.cuda-backends.target.win-64.dependencies] -cupy = "*" +cupy = ">=6.7.0" # jaxlib = { version = "*", build = "cuda12*" } # unavailable pytorch = { version = "*", build = "cuda12*" }