diff --git a/py/ci/Dockerfile.ci b/py/ci/Dockerfile.ci
index 918eaaef07..f2c1500858 100644
--- a/py/ci/Dockerfile.ci
+++ b/py/ci/Dockerfile.ci
@@ -1,4 +1,4 @@
-FROM pytorch/manylinux-builder:cuda11.7
+FROM pytorch/manylinux-builder:cuda11.8
 ARG trt_version
 
 RUN echo -e "Installing with TensorRT ${trt_version}"