diff --git a/pixi.toml b/pixi.toml index 6906fa11..3bab64cb 100644 --- a/pixi.toml +++ b/pixi.toml @@ -195,7 +195,7 @@ jax = ">=0.9.2" system-requirements = { cuda = "12" } [feature.cuda-backends.target.linux.dependencies] -cupy = ">=14.0.1" +cupy = ">=14.1.0" jaxlib = { version = ">=0.9.2", build = "cuda12*" } pytorch = { version = ">=2.10.0", build = "cuda12*" } @@ -205,7 +205,7 @@ pytorch = { version = ">=2.10.0", build = "cuda12*" } # pytorch = { version = "*", build = "cuda12*" } # unavailable [feature.cuda-backends.target.win.dependencies] -cupy = ">=14.0.1" +cupy = ">=14.1.0" # jaxlib = { version = "*", build = "cuda12*" } # unavailable pytorch = { version = ">=2.10.0", build = "cuda12*" }