Skip to content

Commit 1a300f9

Browse files
authored
Update Dockerfile
Switch Dockerfile to CUDA runtime image
1 parent 6b23ccd commit 1a300f9

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

Dockerfile

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
FROM nvidia/cuda:12.4.0-devel-ubuntu22.04
1+
FROM nvidia/cuda:12.8.1-runtime-ubuntu22.04
22

33
# Set environment variables
44
ENV PYTHONDONTWRITEBYTECODE=1
@@ -33,4 +33,4 @@ RUN mkdir -p model_cache reference_audio outputs voices
3333
EXPOSE 8003
3434

3535
# Command to run the application
36-
CMD ["python3", "server.py"]
36+
CMD ["python3", "server.py"]

0 commit comments

Comments
 (0)