Update cloud init scripts

Author: OguzPastirmaci

files/oke-nvme-raid.sh

@@ -43,6 +43,12 @@ fi
 # Determine config for detected device count and RAID level
 count=${#devices[@]}; bs=4; chunk=256
 stride=$((chunk/bs)) # chunk size / block size
+
+# If only 1 device, force RAID level 0
+if [[ $count -eq 1 ]]; then
+  level=0
+fi
+
 eff_count=$count # $level == 0
 if [[ $level == 10 ]]; then eff_count=$((count/2)); fi
 if [[ $level == 5 ]]; then eff_count=$((count-1)); fi

files/oke-ubuntu-cloud-init.sh

@@ -8,6 +8,13 @@ else
     exit 1
 fi
 
+# Disable nvidia-imex.service for GB200 and GB300 shapes for Dynamic Resource Allocation (DRA) compatibility
+SHAPE=$(curl -H "Authorization: Bearer Oracle" -L http://169.254.169.254/opc/v2/instance/shape)
+if [[ "$SHAPE" == BM.GPU.GB200* ]] || [[ "$SHAPE" == BM.GPU.GB300* ]]; then
+    echo "Disabling nvidia-imex.service for shape: $SHAPE"
+    systemctl disable --now nvidia-imex.service && systemctl mask nvidia-imex.service
+fi
+
 case "$ID" in
     ubuntu)
         echo "Detected Ubuntu"