diff --git a/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskfull-install.sh b/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskfull-install.sh
index b5da8d579..410da181e 100755
--- a/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskfull-install.sh
+++ b/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskfull-install.sh
@@ -3,11 +3,11 @@
 ########
 # Set all the variables below
 
-LINUX_DISTRO="rhels7.4"
+LINUX_DISTRO="rhels7.5-alternate"
 LINUX_ARCH="ppc64le"
 
 COMPUTE_NODE="c910f03c01p10"
-SOURCE_DIR="/media/xcat"
+SOURCE_DIR="/install/tmp"
 
 ########
 
@@ -57,6 +57,9 @@ do
 	"dkms-"*".el7.noarch.rpm")
 		DKMS_RPM="${r}"
 		;;
+        "nvidia-driver-local-repo-rhel"*"-"*".${LINUX_ARCH}.rpm")
+                CUDA_RPMS+=("${r}")
+                ;;
 	esac
 done
 
@@ -97,6 +100,8 @@ do
 done
 ########
 
+umask 0022
+
 OSIMAGE_NAME="${LINUX_DISTRO}-${LINUX_ARCH}-install-cudafull"
 OSIMAGE_OTHERPKGDIR="/install/post/otherpkgs/${LINUX_DISTRO}/${LINUX_ARCH}"
 
diff --git a/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskless-install.sh b/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskless-install.sh
index 731f324fc..e8c9030b2 100755
--- a/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskless-install.sh
+++ b/xCAT-test/autotest/testcase/cuda/rhel-cuda-diskless-install.sh
@@ -3,11 +3,11 @@
 ########
 # Set all the variables below
 
-LINUX_DISTRO="rhels7.4"
+LINUX_DISTRO="rhels7.5-alternate"
 LINUX_ARCH="ppc64le"
 
 COMPUTE_NODE="c910f03c01p10"
-SOURCE_DIR="/media/xcat"
+SOURCE_DIR="/install/tmp"
 
 ########
 
@@ -57,6 +57,9 @@ do
 	"dkms-"*".el7.noarch.rpm")
 		DKMS_RPM="${r}"
 		;;
+	"nvidia-driver-local-repo-rhel"*"-"*".${LINUX_ARCH}.rpm")
+		CUDA_RPMS+=("${r}")
+		;;
 	esac
 done
 
@@ -97,6 +100,8 @@ do
 done
 ########
 
+umask 0022
+
 OSIMAGE_NAME="${LINUX_DISTRO}-${LINUX_ARCH}-netboot-cudafull"
 OSIMAGE_OTHERPKGDIR="/install/post/otherpkgs/${LINUX_DISTRO}/${LINUX_ARCH}"
 OSIMAGE_ROOTIMGDIR="/install/netboot/${LINUX_DISTRO}/${LINUX_ARCH}/${OSIMAGE_NAME}"
@@ -123,13 +128,26 @@ ${OSIMAGE_NAME}:
     permission=755
     pkgdir=/install/${LINUX_DISTRO}/${LINUX_ARCH}
     pkglist=/opt/xcat/share/xcat/netboot/rh/compute.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.pkglist
-    postinstall=/opt/xcat/share/xcat/netboot/rh/compute.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.postinstall
+    postinstall=/install/custom/netboot/rh/cudafull.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.postinstall
     profile=compute
     provmethod=netboot
     rootimgdir=${OSIMAGE_ROOTIMGDIR}
 EOF
 [ "$?" -ne "0" ] && echo "Make osimage definition failed." >&2 && exit 1
 
+mkdir -p "/install/custom/netboot/rh"
+(
+	cat "/opt/xcat/share/xcat/netboot/rh/compute.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.postinstall"
+	cat <<-EOF
+
+	cp /install/postscripts/cuda_power9_setup "${OSIMAGE_ROOTIMGDIR}/rootimg/tmp/cuda_power9_setup"
+	chroot "${OSIMAGE_ROOTIMGDIR}/rootimg" /tmp/cuda_power9_setup
+
+	rm -f "${OSIMAGE_ROOTIMGDIR}/rootimg/tmp/cuda_power9_setup"
+	EOF
+) >"/install/custom/netboot/rh/cudafull.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.postinstall"
+chmod 0755 "/install/custom/netboot/rh/cudafull.${LINUX_DISTRO%%.*}.${LINUX_ARCH}.postinstall"
+
 rm -rf "${OSIMAGE_OTHERPKGDIR}"
 mkdir -p "${OSIMAGE_OTHERPKGDIR}"
 for f in "${CUDA_RPMS[@]}"