feat: update dockerfile
This commit is contained in:
@@ -2,7 +2,7 @@
|
|||||||
# Optimized for RTX 5080 GPU deployment
|
# Optimized for RTX 5080 GPU deployment
|
||||||
|
|
||||||
# Use NVIDIA CUDA base image with Python 3.10
|
# Use NVIDIA CUDA base image with Python 3.10
|
||||||
FROM nvidia/cuda:12.8.0-runtime-ubuntu24.04
|
FROM nvidia/cuda:12.9.0-runtime-ubuntu24.04
|
||||||
|
|
||||||
# Set environment variables
|
# Set environment variables
|
||||||
ENV PYTHONUNBUFFERED=1 \
|
ENV PYTHONUNBUFFERED=1 \
|
||||||
|
|||||||
Reference in New Issue
Block a user