Skip to content

Commit 5af5ce1

Browse files
upgrade
1 parent c9fd7a1 commit 5af5ce1

File tree

2 files changed

+335
-240
lines changed

2 files changed

+335
-240
lines changed

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -13,7 +13,7 @@ explicit = true
1313

1414
[[tool.uv.index]]
1515
name = "torch-gpu"
16-
url = "https://download.pytorch.org/whl/cu121"
16+
url = "https://download.pytorch.org/whl/cu128"
1717
explicit = true
1818

1919
# NVIDIA packages (TensorRT, etc.)
@@ -180,7 +180,7 @@ dependencies = [
180180
"pandas<3.0.0,>=2.0.0",
181181
"opencv-python<5.0.0,>=4.7.0",
182182

183-
"torch>=2.2.1,<3.0.0",
183+
"torch>=2.8.0",
184184
"torchvision>=0.17.1,<1.0.0",
185185
"click>=8.1.8",
186186

@@ -217,7 +217,7 @@ yolo = [
217217
]
218218

219219
evolve = [
220-
"ray[tune]==2.49.2",
220+
"ray[tune]>2.52.0",
221221
"plotly==5.19.0",
222222
"bayesian-optimization==2.0.4",
223223
"optuna==3.5.0",

0 commit comments

Comments
 (0)