add script to install AMD drivers

Author: r2k1

parts/linux/cloud-init/artifacts/cse_config.sh

@@ -844,6 +844,24 @@ ensureGPUDrivers() {
     fi
 }
 
+# TODO: this is a temporary ubuntu-only HACK until we get a driver
+ensureAMDGPUDrivers() {
+    echo "Installing AMD GPU drivers"
+
+    # delete amdgpu module from blacklist
+    sudo sed -i '/blacklist amdgpu/d' /etc/modprobe.d/blacklist-radeon-instinct.conf
+
+    # temporary solution, until the driver is available in MCR
+    sudo apt-get update
+    wget https://repo.radeon.com/amdgpu-install/6.3.1/ubuntu/jammy/amdgpu-install_6.3.60301-1_all.deb
+    sudo apt-get install -y ./amdgpu-install_6.3.60301-1_all.deb
+    sudo apt-get update
+    sudo apt-get install -y amdgpu-dkms
+
+    REBOOTREQUIRED=true
+    echo "AMD GPU drivers installed"
+}
+
 disableSSH() {
     systemctlDisableAndStop ssh || exit $ERR_DISABLE_SSH
 }