diff --git a/pipelines/build/dockerFiles/cuda.dockerfile b/pipelines/build/dockerFiles/cuda.dockerfile
index 1f4e4e27b..6307d02db 100644
--- a/pipelines/build/dockerFiles/cuda.dockerfile
+++ b/pipelines/build/dockerFiles/cuda.dockerfile
@@ -1,6 +1,6 @@
 ARG image
 ARG cuda_ver=12.2.0
-ARG cuda_distro=ubuntu20.04
+ARG cuda_distro=ubi8
 
 FROM nvidia/cuda:${cuda_ver}-devel-${cuda_distro} as cuda
 FROM $image