Update nvidia/cuda Docker tag to v12.3.2

IQTLabs · Mar 15, 2024 · a11f2e9 · a11f2e9
1 parent 36aa16f
commit a11f2e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/Dockerfile.cuda-torchserve b/docker/Dockerfile.cuda-torchserve
@@ -1,4 +1,4 @@
-FROM nvidia/cuda:12.1.0-runtime-ubuntu22.04
+FROM nvidia/cuda:12.3.2-runtime-ubuntu22.04
 ARG PIP_CACHE
 ENV DEBIAN_FRONTEND=noninteractive
 WORKDIR /root