Skip to content

Commit b24d7fb

Browse files
committed
ci: do not install GPU versions of pytorch via uv
(assuming CI uses `uv pip install ...`)
1 parent 4018699 commit b24d7fb

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pyproject.toml

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -64,3 +64,15 @@ ignore = [
6464
"E702", # multiple-statements-on-one-line-semicolon
6565
"E731", # lambda-assignment
6666
]
67+
68+
69+
# torch (for CI): use CPU-only whl, no need to install CUDA stuffs
70+
[tool.uv.sources]
71+
torch = [
72+
{ index = "pytorch-cpu" },
73+
]
74+
75+
[[tool.uv.index]]
76+
name = "pytorch-cpu"
77+
url = "https://download.pytorch.org/whl/cpu"
78+
explicit = true

0 commit comments

Comments
 (0)