diff --git a/src/worker/install/common.sh b/src/worker/install/common.sh
index 242c74b..8dba419 100644
--- a/src/worker/install/common.sh
+++ b/src/worker/install/common.sh
@@ -7,7 +7,7 @@ set -e
 distro=`awk -F= '/^NAME/{print $2}' /etc/os-release`
 if [[ $distro =~ "Ubuntu" ]]; then
     apt-get install -y python3-dev
-elif [[ $distro =~ "AlmaLinux" ]]; then
+elif [[ $distro =~ (AlmaLinux|Rocky Linux) ]]; then
     yum install -y python3-devel
 else
 	echo "OS version is not supported"
diff --git a/src/worker/install/nvidia.sh b/src/worker/install/nvidia.sh
index ce3f639..aa6c898 100755
--- a/src/worker/install/nvidia.sh
+++ b/src/worker/install/nvidia.sh
@@ -67,7 +67,7 @@ if [[ $distro =~ "Ubuntu" ]]; then
 		systemctl --now enable nvidia-dcgm
 		systemctl start nvidia-dcgm
 	fi
-elif [[ $distro =~ "AlmaLinux" ]]; then
+elif [[ $distro =~ (AlmaLinux|Rocky Linux) ]]; then
 	dcgm_check=`rpm -qa`
 	if [[ $dcgm_check =~ "datacenter-gpu-manager" ]]; then
 		check_min_dcgm_ver alma_dcgm_install "2.4.4"