From d23b0a0e9a8cefce6826bbff73f29d3808932041 Mon Sep 17 00:00:00 2001
From: Mitchell Riedstra <mitch@riedstra.dev>
Date: Sun, 18 Feb 2024 09:11:30 -0500
Subject: Install rocm on Fedora

---
 distro/fedora.sh | 7 +++++++
 1 file changed, 7 insertions(+)

(limited to 'distro')

diff --git a/distro/fedora.sh b/distro/fedora.sh
index 76e2b01..4b4f089 100755
--- a/distro/fedora.sh
+++ b/distro/fedora.sh
@@ -117,6 +117,13 @@ sh -c 'echo "options v4l2loopback devices=3 video_nr=10,11,12 card_label=\"OBS C
 flatpak remote-add --if-not-exists flathub https://dl.flathub.org/repo/flathub.flatpakrepo
 flatpak install -y flathub com.obsproject.Studio
 
+if lspci | grep -qi 'vga.*amd' ; then
+	sudo usermod -a -G render,video "$SUDO_USER"
+	sudo dnf -y install rocm-hip rocm-smi rocm-opencl-devel
+	printf "\033[1;31m%s\033[0m\n" \
+		"NOTE: YOU MAY NEED TO 'export HIP_VISIBLE_DEVICES=0' IN ORDER FOR PYTORCH TO RUN"
+fi
+
 if lspci | grep -qi 'vga.*nvidia'; then
 	# Nvidia https://rpmfusion.org/Howto/NVIDIA
 	dnf -y install akmod-nvidia vulkan xorg-x11-drv-nvidia-cuda-libs \
-- 
cgit v1.2.3