File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -20,8 +20,10 @@ WORKDIR /app
2020COPY --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
2729COPY --chown=appuser:appuser api ./api
Original file line number Diff line number Diff line change @@ -44,9 +44,9 @@ dependencies = [
4444
4545[project .optional-dependencies ]
4646gpu = [" torch==2.9.0+cu130" ]
47- cpu = [" torch==2.8 .0" ]
47+ cpu = [" torch==2.9 .0" ]
4848rocm = [
49- " torch==2.8 .0+rocm6.4" ,
49+ " torch==2.9 .0+rocm6.4" ,
5050 " pytorch-triton-rocm>=3.2.0" ,
5151]
5252test = [
@@ -87,7 +87,7 @@ explicit = true
8787
8888[[tool .uv .index ]]
8989name = " pytorch-cuda"
90- url = " https://download.pytorch.org/whl/cu129 "
90+ url = " https://download.pytorch.org/whl/cu130 "
9191explicit = true
9292
9393[[tool .uv .index ]]
You can’t perform that action at this time.
0 commit comments