Skip to content

Commit 6d4ed1d

Browse files
committed
Update Dockerfile and pyproject.toml for dependency version upgrades
1 parent 571cd46 commit 6d4ed1d

2 files changed

Lines changed: 7 additions & 5 deletions

File tree

docker/gpu/Dockerfile

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,10 @@ WORKDIR /app
2020
COPY --chown=appuser:appuser pyproject.toml ./pyproject.toml
2121

2222
# Install dependencies with GPU extras
23-
RUN uv venv --python 3.10 && \
24-
uv sync --extra gpu --no-cache
23+
RUN uv venv --python 3.10
24+
25+
RUN uv sync --extra gpu --no-cache
26+
2527

2628
# Copy project files including models
2729
COPY --chown=appuser:appuser api ./api

pyproject.toml

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -44,9 +44,9 @@ dependencies = [
4444

4545
[project.optional-dependencies]
4646
gpu = ["torch==2.9.0+cu130"]
47-
cpu = ["torch==2.8.0"]
47+
cpu = ["torch==2.9.0"]
4848
rocm = [
49-
"torch==2.8.0+rocm6.4",
49+
"torch==2.9.0+rocm6.4",
5050
"pytorch-triton-rocm>=3.2.0",
5151
]
5252
test = [
@@ -87,7 +87,7 @@ explicit = true
8787

8888
[[tool.uv.index]]
8989
name = "pytorch-cuda"
90-
url = "https://download.pytorch.org/whl/cu129"
90+
url = "https://download.pytorch.org/whl/cu130"
9191
explicit = true
9292

9393
[[tool.uv.index]]

0 commit comments

Comments
 (0)