diff --git a/Dockerfile.gpu b/Dockerfile.gpu index 1bb5bd13796fe5710b452d9ef3480a98702239d1..9b00cc7f60dd0e041d1a67b87d62d10ea07b28ff 100644 --- a/Dockerfile.gpu +++ b/Dockerfile.gpu @@ -2,7 +2,7 @@ FROM onerahmet/ffmpeg:n7.1 AS ffmpeg FROM swaggerapi/swagger-ui:v5.9.1 AS swagger-ui -FROM nvidia/cuda:11.8.0-cudnn8-runtime-ubuntu22.04 +FROM nvidia/cuda:12.6.3-cudnn-runtime-ubuntu22.04 ENV PYTHON_VERSION=3.10