public-mirrors/Kokoro-FastAPI
1
0
Fork 0
mirror of https://github.com/remsky/Kokoro-FastAPI.git synced 2025-08-05 16:48:53 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Update torch to 2.7.1 & Cuda 12.8.1 in Docker

Browse source
This commit is contained in:
Miggi 2025-06-27 14:30:47 +02:00
parent 6b1e9d9128
commit 8a55cd5bf5
2 changed files with 4 additions and 4 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
docker/gpu/Dockerfile
View file

@ -1,4 +1,4 @@
FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.0-cudnn-runtime-ubuntu24.04
FROM --platform=$BUILDPLATFORM nvidia/cuda:12.8.1-cudnn-runtime-ubuntu24.04
# Set non-interactive frontend
ENV DEBIAN_FRONTEND=noninteractive

6
pyproject.toml
View file

@ -44,10 +44,10 @@ dependencies = [
[project.optional-dependencies]
gpu = [
"torch==2.6.0+cu124",
"torch==2.7.1+cu128",
]
cpu = [
"torch==2.6.0",
"torch==2.7.1",
]
test = [
"pytest==8.3.5",
@ -79,7 +79,7 @@ explicit = true
[[tool.uv.index]]
name = "pytorch-cuda"
url = "https://download.pytorch.org/whl/cu124"
url = "https://download.pytorch.org/whl/cu128"
explicit = true
[build-system]