@@ -134,7 +134,7 @@ numpy = "=1.22.0"
134
134
# Backends that can run on CPU-only hosts
135
135
# Note: JAX and PyTorch will install CPU variants.
136
136
[tool .pixi .feature .backends .dependencies ]
137
- pytorch = " >=1.13.1 "
137
+ pytorch = " >=2.6.0 "
138
138
dask = " >=2025.3.0"
139
139
numba = " >=0.61.0" # sparse dependency
140
140
llvmlite = " >=0.44.0" # sparse dependency
@@ -166,7 +166,7 @@ system-requirements = { cuda = "12" }
166
166
[tool .pixi .feature .cuda-backends .target .linux-64 .dependencies ]
167
167
cupy = " >=13.4.1"
168
168
jaxlib = { version = " >=0.5.2" , build = " cuda12*" }
169
- pytorch = { version = " >=1.13.1 " , build = " cuda12*" }
169
+ pytorch = { version = " >=2.6.0 " , build = " cuda12*" }
170
170
171
171
[tool .pixi .feature .cuda-backends .target .osx-64 .dependencies ]
172
172
# cupy = "*" # unavailable
@@ -181,7 +181,7 @@ pytorch = { version = ">=1.13.1", build = "cuda12*" }
181
181
[tool .pixi .feature .cuda-backends .target .win-64 .dependencies ]
182
182
cupy = " >=13.4.1"
183
183
# jaxlib = { version = "*", build = "cuda12*" } # unavailable
184
- pytorch = { version = " >=1.13.1 " , build = " cuda12*" }
184
+ pytorch = { version = " >=2.6.0 " , build = " cuda12*" }
185
185
186
186
[tool .pixi .environments ]
187
187
default = { features = [" py313" ], solve-group = " py313" }
0 commit comments