Last active
March 5, 2023 04:13
-
-
Save Spencer-Dawson/7fdb5dd09461b6cece8a99537f381e44 to your computer and use it in GitHub Desktop.
Experimental Ubuntu 22.04 ROCm installer
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| #!/bin/bash | |
| echo "This script is completely untested. " | |
| echo "It's more intended as a general overview of what needs to be done to install ROCm on Ubuntu 22.04 then as an actual installer. " | |
| echo "Use at your own risk. And provide feedback if you have suggested fixes. " | |
| # make sure user wants to continue | |
| read -p "Do you want to continue? [y/N] " -n 1 -r | |
| echo | |
| if [[ ! $REPLY =~ ^[Yy]$ ]]; then | |
| exit 1 | |
| fi | |
| #install rocm in ubuntu 22.04 | |
| # see https://docs.amd.com/bundle/ROCm-Installation-Guide-v5.3/page/How_to_Install_ROCm.html for reference | |
| # Check if the system is running Ubuntu 22.04 | |
| if [ "$(lsb_release -rs)" != "22.04" ]; then | |
| echo "This script is only for Ubuntu 22.04" | |
| exit 1 | |
| fi | |
| # Check for ROCm supported gpu | |
| # todo: check for supported gpu using lspci and grep | |
| if [ "$(lspci | grep -i 'AMD' | grep -i 'Radeon')" == "" ]; then | |
| echo "No supported AMD GPU found. Exiting..." | |
| exit 1 | |
| fi | |
| # Ask user to specify the version of ROCm to install | |
| echo "Which version of ROCm do you want to install?" | |
| echo "1) ROCm 5.4" | |
| echo "2) ROCm 5.3" | |
| read -p "Input> " rocm_version | |
| rocm_version=${rocm_version:0:1} | |
| # Convert rocm_version to the actual version number | |
| if [ "$rocm_version" == "1" ]; then | |
| rocm_version="5.4" | |
| elif [ "$rocm_version" == "2" ]; then | |
| rocm_version="5.3" | |
| else | |
| echo "Invalid choice. Exiting..." | |
| exit 1 | |
| fi | |
| # make a tmp directory to install ROCm | |
| mkdir -p /tmp/rocm || exit 1 | |
| cd /tmp/rocm || exit 1 | |
| sudo apt update || exit 1 | |
| # Install ROCm5.4 if user specified | |
| if [ "$rocm_version" == "5.4" ]; then | |
| echo "Installing ROCm 5.4" | |
| wget https://repo.radeon.com/amdgpu-install/5.4.3/ubuntu/jammy/amdgpu-install_5.4.50403-1_all.deb || exit 1 | |
| sudo apt-get install ./amdgpu-install_5.4.50403-1_all.deb || exit 1 | |
| fi | |
| # Install ROCm5.3 if user specified | |
| if [ "$rocm_version" == "5.3" ]; then | |
| echo "Installing ROCm 5.3" | |
| wget https://repo.radeon.com/amdgpu-install/5.3/ubuntu/jammy/amdgpu-install_5.3.50300-1_all.deb || exit 1 | |
| sudo apt install ./amdgpu-install_5.3.50300-1_all.deb || exit 1 | |
| exit 1 | |
| fi | |
| # ensure amdgpu-install is configured properly for ROCm usecase | |
| sudo amdgpu-install --usecase=hiplibsdk,rocm | |
| # cleanup tmp files | |
| cd ~ | |
| rm -rf /tmp/rocm | |
| #add user to video group | |
| sudo usermod -a -G video $USER | |
| echo "Installation complete. Reboot your system to complete the installation." |
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment