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

Merge pull request #354 from MiggiV2/master

Browse source 
Update PyTorch to 2.7.1+cu128 for RTX 50 Series GPU Support
This commit is contained in:
Fireblade2534 2025-07-04 18:50:59 -04:00 • committed by GitHub
commit 29066f7c9f
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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]