diff --git a/api/Containerfile.cuda.ubuntu b/api/Containerfile.cuda.ubuntu index 26c10cd95..50706f76e 100644 --- a/api/Containerfile.cuda.ubuntu +++ b/api/Containerfile.cuda.ubuntu @@ -1,4 +1,4 @@ -FROM docker.io/nvidia/cuda:11.7.1-runtime-ubuntu20.04 +FROM docker.io/nvidia/cuda:12.3.2-runtime-ubuntu20.04 WORKDIR /onnx-web/api