Welcome to this repository, where I share my notes and insights on setting up multiple AMD GPUs on Ubuntu for AI development. This initiative stems from the noticeable gap in resources and discussions around AMD GPU setups for AI, as most online documentation and forums predominantly focus on Nvidia GPUs. This repository aims to bridge that gap, providing an overview and step-by-step guide based on my personal experience and research.
This repository contains notes, guides, and scripts that I've compiled during my setup process. While my setup specifically involves Ubuntu 22.04, kernel 6.5, with 2 AMD RX 7900 XTX GPUs, the information provided here should be applicable or easily adaptable to similar configurations. Here's what to expect:
- Installation Guides: Detailed steps on installing and configuring the necessary drivers and tools for AMD GPUs on Ubuntu.
- Troubleshooting Tips: Solutions to common issues that might arise during the setup process.
- Performance Optimization: Tips on optimizing your AMD GPU setup for better performance in AI development tasks.
- Useful Resources: A curated list of resources that I found invaluable during my setup process.
The notes and guides in this repository are based on my personal experience and research. While I strive to provide accurate and up-to-date information, I cannot guarantee that everything will work perfectly for every setup. Always back up your data and proceed with caution.
-
PCIe® slots connected to the GPU must have identical PCIe lane width or bifurcation settings, and support PCIe 3.0 Atomics.
-
Only use PCIe slots connected by the CPU and to avoid PCIe slots connected via chipset. Refer to product-specific motherboard documentation for PCIe electrical configuration.
-
Ensure the PSU has sufficient wattage to support multiple GPUs.
-
Enable either iGPU or Discrete GPU
Read more at: https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/native_linux/mgpu.html
If you need to install an additional kernel, read:
https://github.com/eliranwong/MultiAMDGPU_AIDev_Ubuntu/blob/main/Install_Ubuntu_Kernel.md
Add current user to groups 'render' and 'video'
sudo usermod -a -G render,video $LOGNAME
To add all future users to the video and render groups by default, run the following commands:
echo 'ADD_EXTRA_GROUPS=1' | sudo tee -a /etc/adduser.conf
echo 'EXTRA_GROUPS=video' | sudo tee -a /etc/adduser.conf
echo 'EXTRA_GROUPS=render' | sudo tee -a /etc/adduser.conf
Explanation
For AMD GPUs on Linux, the groups you might need to add users to for proper GPU access are similar to those for NVIDIA GPUs. Here are the key groups:
- video: This group grants access to video devices and may include GPU devices.
- render: As previously mentioned, this group allows access to GPU rendering devices.
When using AMD GPUs, especially with ROCm (Radeon Open Compute), you may also need to add users to these groups to ensure they have the necessary permissions to access the GPU for computing tasks. The ROCm documentation specifically mentions adding users to both the render and video groups to set the correct permissions.
To add a user to these groups, you can use the following command:
sudo usermod -a -G render,video username
Replace username with the actual username of the user you want to add to the groups. After adding the user to these groups, they should have the necessary permissions to access the GPU resources on your system. It's always a good practice to log out and log back in or reboot the system to ensure the changes take effect.
If you're using specific AMDGPU control applications or tools, they might have their own group requirements or recommendations, so it's a good idea to check the documentation for those tools as well1. Remember, managing user access to GPUs is an important aspect of system administration, especially in multi-user environments or when dealing with sensitive compute tasks.
Version 6.1.3 is preferred, as it officaillly supports AMD Radeon™ 7000 series GPUs:
AMD has expanded support for Machine Learning Development on RDNA™ 3 GPUs with Radeon™ Software for Linux 24.10.3 with ROCm™ 6.1.3!
Read more at: https://rocm.docs.amd.com/projects/radeon/en/latest/index.html
amdgpu-install --uninstall
sudo apt remove --purge amdgpu-install
sudo apt update
sudo apt install -y libstdc++-12-dev
wget https://repo.radeon.com/amdgpu-install/6.1.3/ubuntu/jammy/amdgpu-install_6.1.60103-1_all.deb
sudo apt install ./amdgpu-install_6.1.60103-1_all.deb
sudo amdgpu-install --usecase=graphics,multimedia,multimediasdk,rocm,rocmdev,rocmdevtools,lrt,opencl,openclsdk,hip,hiplibsdk,openmpsdk,mllib,mlsdk --no-dkms -y
For more options of use cases, read https://rocm.docs.amd.com/projects/install-on-linux/en/latest/how-to/amdgpu-install.html#use-cases
To install ROCm inside a container, read: Read https://github.com/eliranwong/incus_container_gui_setup
Solution: Add "iommu=pt" to GRUB_CMDLINE_LINUX_DEFAULT in /etc/default/grub.
For example,
sudo nano /etc/default/grub
Changed from:
GRUB_CMDLINE_LINUX_DEFAULT="quiet splash"
To:
GRUB_CMDLINE_LINUX_DEFAULT="quiet splash iommu=pt"
Update grub
sudo update-grub
Restart to make changes effective:
sudo reboot
To verify, run:
rocminfo
If you use Xorg instead of Wayland and have the issue where the mouse cursor is invisible, you can try to create the file /etc/X11/xorg.conf.d/99-modesetting.conf with the following content:
Section "Device"
Identifier "modesetting"
Driver "modesetting"
EndSection
To avoid a known bug in underlying ROCm/HIP runtime, disable the integrated GPU.
The file /etc/default/grub
is a configuration file for the GRUB bootloader on Ubuntu. The line GRUB_CMDLINE_LINUX_DEFAULT
in this file is used to pass arguments to the Linux kernel at boot time. To disable an integrated GPU system-wide, you can do so by adding a specific argument to this line. The argument you need is pci-stub.ids=\<DEVICE_VENDOR\>:\<DEVICE_CODE\>
, where \<DEVICE_VENDOR\>:\<DEVICE_CODE\>
is the vendor and device code of your GPU.
Check the vendor and device code with 'lspci', e.g.:
lspci -k | grep -A 2 -i "VGA"
For example, the following output tells that the vendor and device code is '1f66:0001':
05:00.0 VGA compatible controller: Advanced Micro Devices, Inc. [AMD/ATI] Rembrandt (rev c7)
Subsystem: Device 1f66:0001
Kernel driver in use: amdgpu
Edit GRUB bootloader configuration file:
sudo nano /etc/default/grub
GRUB_CMDLINE_LINUX_DEFAULT="quiet splash pci-stub.ids=1f66:0001"
Update GRUB for the changes to take effect
sudo update-grub
Explanation
The GRUB_CMDLINE_LINUX_DEFAULT line in the GRUB configuration file is used to set the default kernel boot parameters. These parameters are passed to the Linux kernel at boot time and can be used to control various aspects of the system’s behavior.
The pci-stub.ids=\<DEVICE_VENDOR\>:\<DEVICE_CODE\>
argument doesn't exactly "disable" the integrated graphics card, but rather it prevents the Linux kernel from loading a specific driver for that device during the boot process.
Here's how it works:
-
Each hardware device in your computer, including your integrated graphics card, has a unique vendor and device ID. This ID is used by the operating system to identify the device and load the appropriate driver for it.
-
When you pass
pci-stub.ids=\<DEVICE_VENDOR\>:\<DEVICE_CODE\>
to the kernel at boot time, you're telling the kernel to reserve the specified device (in this case, your integrated graphics card) for thepci-stub
driver. -
The
pci-stub
driver is a "dummy" driver that doesn't do anything—it simply claims the device and prevents other drivers from being able to use it. This is useful in situations where you want to prevent the operating system from interacting with a device, such as when you're setting up a device for pass-through to a virtual machine.
So, while the integrated graphics card is still physically present and powered on, the operating system is unable to interact with it because the pci-stub
driver has claimed it. This effectively "disables" the integrated graphics card from the operating system's perspective.
Remember, this is a low-level operation that can have significant effects on your system, so it should be done with caution. Always make sure to consult the relevant documentation or seek expert advice if you're unsure. And don't forget to run sudo update-grub
and reboot your system after making changes to the GRUB configuration file.
Configure Vulkan to use AMD graphics card:
- Edit the file `/etc/default/grub` and in the line that reads `GRUB_CMDLINE_LINUX_DEFAULT`, add: `radeon.cik_support=0 amdgpu.cik_support=1 radeon.si_support=0 amdgpu.si_support=1`.
- Create a new file `/etc/modprobe.d/amdgpu.conf` and add the following lines to it:
```
options amdgpu si_support=1
options amdgpu cik_support=1
```
- Create a new file `/etc/modprobe.d/radeon.conf` and add the following lines to it:
```
options radeon si_support=0
options radeon cik_support=0
```
- Create a new file `/etc/modprobe.d/blacklist.conf` and add the following lines to it:
```
blacklist radeon
```
- Run `sudo update-grub` and then reboot your system.
With GUI:
- Launch Settings application
- Select About > Graphics
With CLI:
rocminfo
Look for information like:
*******
Agent 2
*******
Name: gfx1100
Uuid: GPU-b54ca445df90862b
Marketing Name: Radeon RX 7900 XTX
Vendor Name: AMD
Feature: KERNEL_DISPATCH
Profile: BASE_PROFILE
Float Round Mode: NEAR
Max Queue Number: 128(0x80)
Queue Min Size: 64(0x40)
Queue Max Size: 131072(0x20000)
Queue Type: MULTI
Node: 1
*******
Agent 3
*******
Name: gfx1100
Uuid: GPU-2ff163adb661d5fb
Marketing Name: Radeon RX 7900 XTX
Vendor Name: AMD
Feature: KERNEL_DISPATCH
Profile: BASE_PROFILE
Float Round Mode: NEAR
Max Queue Number: 128(0x80)
Queue Min Size: 64(0x40)
Queue Max Size: 131072(0x20000)
Queue Type: MULTI
Node: 2
In this case, there are two GPUs, which are referred as device 0 and 1 later.
Modify the values to suit your cases.
The following examples assume:
-
ROCm version 6.1.3 installed
-
No integrated GPU
-
Two AMD RX 7900 XTX installed.
Note: You may run rocm-smi
to find the mapping information of node numbers to devices numbers.
I use my case as an example:
Remarks:
- The following settings assumes
/opt/rocm
points to/opt/rocm-6.1.3
. - Modify the values of ROCR_VISIBLE_DEVICES to your own ones.
# rocm
export GFX_ARCH=gfx1100
export HCC_AMDGPU_TARGET=gfx1100
export CUPY_INSTALL_USE_HIP=1
export ROCM_VERSION=6.1
export ROCM_HOME=/opt/rocm
export LD_LIBRARY_PATH=/usr/include/vulkan:/opt/rocm/include:/opt/rocm/lib:$LD_LIBRARY_PATH
export PATH=/home/eliran/.local/bin:/opt/rocm/bin:/opt/rocm/llvm/bin:$PATH
export HSA_OVERRIDE_GFX_VERSION=11.0.0
export ROCR_VISIBLE_DEVICES=GPU-b54ca445df90862b,GPU-2ff163adb661d5fb
export GPU_DEVICE_ORDINAL=0,1
export HIP_VISIBLE_DEVICES=0,1
export CUDA_VISIBLE_DEVICES=0,1
export LLAMA_HIPLAS=0,1
export DRI_PRIME=1
export OMP_DEFAULT_DEVICE=1
# vulkan
export GGML_VULKAN_DEVICE=0,1
export GGML_VK_VISIBLE_DEVICES=0,1
export VULKAN_SDK=/usr/share/vulkan
export VK_LAYER_PATH=$VULKAN_SDK/explicit_layer.d
ROCM_HOME - tells AI libraries where ROCM is stored; typically somewhere in /opt, e.g.:
export ROCM_HOME=/opt/rocm-6.1.3
Explanation
ROCM_HOME
is an environment variable in Linux that is used to specify the location of the ROCm (Radeon Open Compute) software on your system. ROCm is a set of open-source libraries and tools that are used to create high performance, machine learning applications on AMD GPUs.
When you install ROCm, it is typically installed in a directory under /opt
. For example, if you installed version 6.1.3 of ROCm, it might be installed in /opt/rocm-6.1.3
.
The export
command in Linux is used to set environment variables. So, when you run the command export ROCM_HOME=/opt/rocm-6.1.3
, you are telling the system "Whenever I refer to ROCM_HOME
, I actually mean /opt/rocm-6.1.3
".
This is useful because many AI libraries that use ROCm will look for the ROCM_HOME
environment variable to know where to find the ROCm software. By setting ROCM_HOME
, you ensure that these libraries can find and use ROCm correctly.
LD_LIBRARY_PATH - loader library path; typically this is set to $ROCM_HOME/lib. An indication you’re missing this flag is if you import pytorch and see an error like undefined reference to...
export LD_LIBRARY_PATH=/opt/rocm-6.1.3/lib:$LD_LIBRARY_PATH
export PATH=/opt/rocm-6.1.3/bin:/opt/rocm-6.1.3/opencl/bin:$PATH
Explanation
LD_LIBRARY_PATH
is an environment variable that specifies a list of directories where the dynamic linker should look for dynamically linked libraries. When a program is launched, the dynamic linker checks the LD_LIBRARY_PATH
to find the libraries that the program needs to run.
In this example, LD_LIBRARY_PATH
is being set to include the /opt/rocm-6.1.3/lib
directory. This is likely where the ROCm (Radeon Open Compute) libraries are installed. If you're trying to use PyTorch and it's built with ROCm support, it will need to know where these libraries are. If LD_LIBRARY_PATH
doesn't include the ROCm library directory, you might see errors like "undefined reference to..." when you try to import PyTorch.
The command export LD_LIBRARY_PATH=/opt/rocm-6.1.3/lib:$LD_LIBRARY_PATH
is adding /opt/rocm-6.1.3/lib
to your existing LD_LIBRARY_PATH
.
The PATH
environment variable is similar, but it's used to tell the shell where to look for executable files. The command export PATH=/opt/rocm-6.1.3/bin:/opt/rocm-6.1.3/opencl/bin:$PATH
is adding the /opt/rocm-6.1.3/bin
and /opt/rocm-6.1.3/opencl/bin
directories to your PATH
. This means that when you type a command, the shell will also look in these directories to find it.
HSA_OVERRIDE_GFX_VERSION - workaround for software that doesn’t yet fully support the installed gpu.
rocminfo | grep gfx
For example, gfx1100 used by AMD RX 7900 XTX:
export HSA_OVERRIDE_GFX_VERSION=11.0.0
Explanation
HSA_OVERRIDE_GFX_VERSION: This is an environment variable used as a workaround for software that doesn’t yet fully support the installed GPU. It's used to override the Graphics/Compute version. For example, if you have a GPU that is not yet fully supported by the ROCm software (like the AMD Radeon RX 7900 XTX which uses gfx1100), you can set this environment variable to tell the ROCm software to treat your GPU as if it were a different, fully supported version.
rocminfo | grep gfx: This is a command-line instruction. rocminfo
is a tool that provides information about the HSA (Heterogeneous System Architecture) system attributes and agents. The grep gfx
part of the command filters the output of rocminfo
to only show lines that contain 'gfx', which are the lines that tell you the version of the AMD GCN ISA or architecture names.
export HSA_OVERRIDE_GFX_VERSION=11.0.0: This is a command that sets the HSA_OVERRIDE_GFX_VERSION
environment variable to '11.0.0'. This tells the ROCm software to treat your GPU as if it were a gfx1100, even if it's actually a different version. This can be useful if your GPU is not yet fully supported by the ROCm software.
In summary, these commands and variables are used to help ensure compatibility between your GPU and the ROCm software, even if your GPU is not yet fully supported. They allow you to override the reported version of your GPU so that the ROCm software treats it as a fully supported version. This can be particularly useful when working with newer GPUs like the AMD Radeon RX 7900 XTX. Please note that while this can enable you to use the ROCm software with unsupported GPUs, it may not provide optimal performance or full functionality. It's always best to check the official ROCm documentation or the GPU manufacturer's documentation for the most accurate and up-to-date information.
Remarks: Set HSA_OVERRIDE_GFX_VERSION=10.3.0 for 680M, and HSA_OVERRIDE_GFX_VERSION=11.0.0 for 780M.
ROCR_VISIBLE_DEVICES - device indices or UUIDs that will be exposed to applications, e.g.:
export ROCR_VISIBLE_DEVICES=0,1
Remarks: Tough documents state that device indices are accepted, but device indeces does not work in my case. I have to use UUIDs, in order for my system to detect the GPUs correctly. UUIDs of individual GPUs can be found with command 'rocminfo'.
Explanation
ROCR_VISIBLE_DEVICES
is an environment variable used in the ROCm (Radeon Open Compute) software stack. It specifies which GPU devices will be exposed to applications.
Device Indices or UUIDs - These are identifiers for the GPUs in your system. A device index is a numerical value assigned to each GPU, starting from 0. A UUID (Universally Unique Identifier) is a unique string that can also be used to identify a GPU.
You set this environment variable to a list of device indices or UUIDs that you want to expose to applications. For example, export ROCR_VISIBLE_DEVICES=0,1
means that only the first and second GPUs (indices start from 0) will be visible to applications.
Applications running on the ROCm platform will only be able to see and use the GPUs specified in ROCR_VISIBLE_DEVICES
. Other GPUs in the system will be hidden from these applications.
This feature is useful for isolating GPU resources, especially in multi-GPU systems. For instance, you might want certain applications to only use specific GPUs, while other GPUs are reserved for different tasks.
GPU_DEVICE_ORDINAL - devices indices exposed to OpenCL and HIP applications, e.g.:
export GPU_DEVICE_ORDINAL=0,1
Explanation
GPU_DEVICE_ORDINAL
is an environment variable used in both OpenCL and HIP (Heterogeneous-Compute Interface for Portability) applications. It's used to control the visibility of devices to these applications, similar to HIP_VISIBLE_DEVICES
.
This specific environment variable, GPU_DEVICE_ORDINAL
, is used to control which devices are available for OpenCL and HIP programs to use when they are run.
Device Indices are the numerical identifiers assigned to each device. In a system with multiple GPUs, each GPU is assigned a unique device index.
The value 0,1
means that the first and second devices (as device indices start from 0) are visible to OpenCL and HIP applications. So, if you have two GPUs in your system, both will be available for use by these programs.
HIP_VISIBLE_DEVICES - Device indices exposed to HIP applications, e.g.:
export HIP_VISIBLE_DEVICES=0,1
Explanation
HIP_VISIBLE_DEVICES
is an environment variable used in HIP (Heterogeneous-Compute Interface for Portability) applications. HIP is a part of AMD's ROCm (Radeon Open Compute) platform designed to ease the task of porting CUDA applications to AMD's hardware.
This specific environment variable, HIP_VISIBLE_DEVICES
, is used to control the visibility of devices to HIP applications. It determines which devices are available for HIP programs to use when they are run.
Device Indices are the numerical identifiers assigned to each device. In a system with multiple GPUs, each GPU is assigned a unique device index.
The value 0,1
means that the first and second devices (as device indices start from 0) are visible to HIP applications. So, if you have two GPUs in your system, both will be available for use by HIP programs.
CUDA_VISIBLE_DEVICES - provided for CUDA compatibility, e.g.:
export CUDA_VISIBLE_DEVICES=0,1
Explanation
CUDA_VISIBLE_DEVICES
is an environment variable in Linux that CUDA applications use to control which GPUs they can use. This variable is provided for CUDA compatibility.
When you run a CUDA application, it can see all the GPUs in your system by default. However, there might be situations where you want to limit an application to only use certain GPUs. This is where CUDA_VISIBLE_DEVICES
comes into play.
The export CUDA_VISIBLE_DEVICES=0,1
command is an example of how you can use this environment variable. Here's what it does:
export
: This is a command in Linux that sets environment variables.CUDA_VISIBLE_DEVICES
: This is the environment variable we're setting. It controls which GPUs a CUDA application can use.0,1
: This is the value we're setting forCUDA_VISIBLE_DEVICES
. The numbers correspond to the IDs of the GPUs.0,1
means the application can use GPU 0 and GPU 1.
So, export CUDA_VISIBLE_DEVICES=0,1
means "set the CUDA_VISIBLE_DEVICES
environment variable to 0,1
". After running this command, any CUDA application you run will only be able to see and use GPU 0 and GPU 1, even if there are more GPUs in the system.
This can be particularly useful in multi-GPU systems, where you might want to reserve certain GPUs for specific tasks or users. By setting CUDA_VISIBLE_DEVICES
, you can control the resources that different applications or users can access.
Please note that the GPU IDs are not necessarily fixed, they can change based on the system configuration, GPU topology, or after a system reboot. So, it's always a good idea to verify the GPU IDs before setting CUDA_VISIBLE_DEVICES
. You can do this using the nvidia-smi
command, which provides information about the GPUs in your system, including their IDs.
LLAMA_HIPLAS - applicable to Llama.cpp setup
export LLAMA_HIPLAS=0,1
Explanation
The LLAMA_HIPLAS=0,1 setting is likely related to the configuration of the LLaMA (Large Language Model) library in the llama.cpp project. However, the exact meaning of this setting is not directly available in the search results.
DRI_PRIME - optional in case with no integrated GPU
export DRI_PRIME=0
Explanation
DRI_PRIME
is an environment variable used in Linux systems to manage hybrid graphics. Hybrid graphics are found on recent desktops and laptops, where there are two graphics cards: an integrated one (usually Intel) and a discrete one (like NVIDIA or AMD Radeon). The integrated card is used for regular tasks to save power, while the discrete card is used for GPU-intensive applications like gaming or 3D rendering.
When you run a command with DRI_PRIME=1
, it tells the system to use the discrete GPU for that particular application. For example, if you want to run Firefox using the discrete GPU, you would use the command DRI_PRIME=1 firefox
.
On the other hand, export DRI_PRIME=0
sets the DRI_PRIME
environment variable to 0
for the entire session or script where the command is run. This means that the integrated GPU (which is usually less powerful but more energy-efficient) will be used for all applications run in that session or script.
Please note that the actual GPU used can depend on your specific system configuration. You can check which GPU an application is using with the command glxinfo | grep "OpenGL renderer"
.
OMP_DEFAULT_DEVICE - default device used for OpenMP target offloading, e.g.:
export OMP_DEFAULT_DEVICE=1
Explanation
OMP_DEFAULT_DEVICE
is an environment variable used in the context of OpenMP, a parallel programming model. This variable is used to specify the default device for OpenMP target offloading.
In OpenMP, "offloading" refers to the process of transferring computation from the host (usually a CPU) to a device (usually a GPU or another accelerator). This is particularly useful in high-performance computing scenarios where you want to leverage the power of GPUs for certain parts of your computation.
The value of OMP_DEFAULT_DEVICE
is an integer that corresponds to the device ID. Device IDs usually start from 0 and increment for each additional device. So, if you have two devices and you want to use the second device as the default for offloading, you would set OMP_DEFAULT_DEVICE="1"
(since we start counting from 0).
It's not uncommon to skip the first device for offloading. The first device (device 0) could be reserved for other tasks, such as rendering graphics in a desktop environment. Offloading compute-intensive tasks to other devices can help ensure that the system remains responsive. However, this can vary based on the specific system configuration and the requirements of the application. It's always a good idea to check the documentation for your specific hardware and software setup to understand the best practices for your situation.
GGML_VULKAN_DEVICE & GGML_VK_VISIBLE_DEVICES - use Llama.cpp via Vulkan backend
export GGML_VULKAN_DEVICE=0,1
export GGML_VK_VISIBLE_DEVICES=0,1
Explanation
Read more at:
-
https://rocmdocs.amd.com/en/latest/conceptual/gpu-isolation.html#environment-variables
-
https://rocmdocs.amd.com/projects/HIP/en/develop/how-to/debugging.html#useful-environment-variables
This is optional if you don't use vulkan.
sudo apt install vulkan-tools libvulkan-dev vulkan-validationlayers vulkan-validationlayers-dev
To verify:
vulkaninfo
e.g.
locate explicit_layer.d # /usr/share/vulkan/explicit_layer.d
To set related Vulkan SDK environment variables:
export VULKAN_SDK=/usr/share/vulkan
export VK_LAYER_PATH=$VULKAN_SDK/explicit_layer.d
Reference: https://github.com/ggerganov/llama.cpp#vulkan
Install ROCm before installing MIGraphX. To install MIGraphX
sudo apt update && sudo apt install -y migraphx
Header files and libraries are installed under /opt/rocm-<version>, where <version> is the ROCm version.
Read: https://github.com/ROCm/AMDMIGraphX#amd-migraphx
Use python version 3.10.x, to work with wheel files available at https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/
To install a specific version with pyenv, read https://github.com/eliranwong/MultiAMDGPU_AIDev_Ubuntu/blob/main/ubuntu_desktop/basic.md#pyenv
sudo apt update
sudo apt install -y make build-essential python3 python-setuptools libjpeg-dev python3-pip python3-dev python3-venv libssl-dev libffi-dev libnss3 zlib1g-dev libbz2-dev libreadline-dev libsqlite3-dev wget curl llvm libncurses5-dev xz-utils tk-dev libxml2-dev libxmlsec1-dev libffi-dev liblzma-dev python3-wheel python3-wheel-whl twine
To make it clean, uninstall old copies, if any
pip uninstall torch torchaudio torchvision cupy spacy numpy protobuf -y
python3 -m venv ai
source ai/bin/activate
pip3 install --upgrade pip wheel setuptools
Run or re-run this line if any conflicts about versions of numpy / protobuf:
pip install numpy==1.26.4 protobuf==4.25.3
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torch-2.1.2%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/torchvision-0.16.1%2Brocm6.1.3-cp310-cp310-linux_x86_64.whl
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/pytorch_triton_rocm-2.1.0%2Brocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl
pip3 install torch-2.1.2+rocm6.1.3-cp310-cp310-linux_x86_64.whl torchvision-0.16.1+rocm6.1.3-cp310-cp310-linux_x86_64.whl pytorch_triton_rocm-2.1.0+rocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl
To verify:
python3 -c 'import torch' 2> /dev/null && echo 'Success' || echo 'Failure'
python3 -c 'import torch; print(torch.cuda.is_available())'
python3 -c "import torch; print(f'device name [0]:', torch.cuda.get_device_name(0))"
python3 -m torch.utils.collect_env
Alternately, run:
python3
>>> import torch
>>> torch.__version__
'2.1.2+rocm6.1.3'
>>> torch.cuda.is_available()
True
>>> torch.cuda.device_count()
2
>>> torch.cuda.get_device_properties(0).total_memory
25753026560
>>> torch.cuda.get_device_properties(1).total_memory
25753026560
>>> torch.cuda.current_device()
0
>>> torch.cuda.get_device_name(torch.cuda.current_device())
'Radeon RX 7900 XTX'
Read more at https://pytorch.org/get-started/locally/#linux-verification
Install migraphx
FIRST!
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/onnxruntime_rocm-1.17.0-cp310-cp310-linux_x86_64.whl
pip install onnxruntime_rocm-1.17.0-cp310-cp310-linux_x86_64.whl
To verify:
python3 -c "import onnxruntime; print(onnxruntime.get_available_providers())"
Expected output:
['MIGraphXExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']
To use MiGraphX ExecutionProvider:
Read: https://onnxruntime.ai/docs/execution-providers/MIGraphX-ExecutionProvider.html
providers = [
'MIGraphXExecutionProvider',
'CPUExecutionProvider',
]
To use ROCm ExecutionProvider:
Read: https://onnxruntime.ai/docs/execution-providers/ROCm-ExecutionProvider.html
providers = [
'ROCMExecutionProvider',
'CPUExecutionProvider',
]
Option: user_compute_stream
providers = [("ROCMExecutionProvider", {"device_id": torch.cuda.current_device(), "user_compute_stream": str(torch.cuda.current_stream().cuda_stream)})]
wget https://repo.radeon.com/rocm/manylinux/rocm-rel-6.1.3/tensorflow_rocm-2.15.1-cp310-cp310-manylinux_2_28_x86_64.whl
pip3 install tensorflow_rocm-2.15.1-cp310-cp310-manylinux_2_28_x86_64.whl
To verify:
python3 -c 'import tensorflow' 2> /dev/null && echo 'Success' || echo 'Failure'
Export required variables, if you haven't:
export CUPY_INSTALL_USE_HIP=1
export ROCM_HOME=/opt/rocm
export HCC_AMDGPU_TARGET=gfx1100
Install from source:
git clone https://github.com/cupy/cupy.git
cd cupy
git checkout rocm-ci-6.1
git submodule update --init
pip install git+https://github.com/ROCmSoftwarePlatform/hipify_torch.git
pip install .
To fix cicular import error, run again:
pip3 install torch-2.1.2+rocm6.1.3-cp310-cp310-linux_x86_64.whl torchvision-0.16.1+rocm6.1.3-cp310-cp310-linux_x86_64.whl pytorch_triton_rocm-2.1.0+rocm6.1.3.4d510c3a44-cp310-cp310-linux_x86_64.whl
To verify:
python3 -c "import cupy; print(cupy.__version__)"
With pytorch & cupy installed first, run:
pip install spacy
Piper-tts is a good offline tts engine that supports Linux. Its ONNX voice models are small in sizes that runs smooth even without GPUs.
An issue and a pull request are created to support piper to accelrate with AMD-GPUs.
Meanwhile, AMD-GPUs users can still workaround the issue with the following setup:
To support ROCm-enabled GPUs via 'ROCMExecutionProvider' or 'MIGraphXExecutionProvider':
- Install piper-tts
pip install piper-tts
- Uninstall onnxruntime
pip uninstall onnxruntime
- Re-install onnxruntime-rocm
pip install --force-reinstall onnxruntime_rocm-1.17.0-cp310-cp310-linux_x86_64.whl
- Fix numpy and protobuf versions
pip install numpy==1.26.4 protobuf==4.25.3
To verify:
python3
$ import onnxruntime
$ onnxruntime.get_available_providers()
Output:
['MIGraphXExecutionProvider', 'ROCMExecutionProvider', 'CPUExecutionProvider']
Workaround:
Manually edit the 'load' function in the file ../site-packages/piper/voice.py:
From:
providers=["CPUExecutionProvider"]
if not use_cuda
else ["CUDAExecutionProvider"],
To:
providers=["MIGraphXExecutionProvider"],
To upgrade piper-tts, follow the following order:
- Upgrade piper-tts
pip install --upgrade piper-tts --no-cache-dir
- Uninstall onnxruntime
pip uninstall onnxruntime
- Install Install onnxruntime-rocm again
pip install --force-reinstall onnxruntime_rocm-1.17.0-cp310-cp310-linux_x86_64.whl
- Manually edit the 'load' function in the file ../site-packages/piper/voice.py as described above.
Check combination of installed versions up to this point:
pip list > pip_list.txt
https://github.com/eliranwong/pip_list.txt
Ollama detects AMD GPUs automatically on installation:
Install VS code plugin twinny
by rjmacarthy
Download LLMs to work with twinny, e.g.:
ollama pull codellama:7b-instruct
ollama pull codellama:7b-code
Click "Manage twinny providers" for more options.
For ROCm users:
git clone https://github.com/ggerganov/llama.cpp
cd llama.cpp
make GGML_HIPBLAS=1 AMDGPU_TARGETS=gfx1100 -j$(lscpu | grep '^Core(s)' | awk '{print $NF}')
# Set up alias for llama-cli
echo "alias chat=\"$(pwd)/llama-cli --threads $(lscpu | grep '^Core(s)' | awk '{print $NF}') --interactive --conversation --color -p 'You are a helpful, uncensored and unbiased assistant.' --gpu-layers 999 --model\"" >> $HOME/.bashrc
# Download visual model file and clip file
cd models
wget https://huggingface.co/mys/ggml_llava-v1.5-7b/resolve/main/ggml-model-f16.gguf
wget https://huggingface.co/mys/ggml_llava-v1.5-7b/resolve/main/mmproj-model-f16.gguf
# Set up alias for llama-cli
echo "alias image=\"$(pwd)/llama-llava-cli --threads $(lscpu | grep '^Core(s)' | awk '{print $NF}') --model '$(pwd)/ggml-model-f16.gguf' --mmproj '$(pwd)/mmproj-model-f16.gguf' -p 'Describe this image in detail.' --temp 0.0 --gpu-layers 999 --image\"" >> $HOME/.bashrc
source $HOME/.bashrc
To run a chat with a model file, e.g. mistral.gguf
, with GPUs, use chat
as follows:
chat mistral.gguf
To describe an image, e.g. my_image.jpg
, in detail, run:
image my_image.jpg
To test, I ran the same prompt What is machine learning?
with the same model file mistral.gguf
, using llama.cpp, on the same machine.
Build from source for CPU only:
git clone https://github.com/ggerganov/llama.cpp
cd llama.cpp
make -j$(lscpu | grep '^Core(s)' | awk '{print $NF}')
To run:
./llama-cli -t $(lscpu | grep '^Core(s)' | awk '{print $NF}') --temp 0 -m ../mistral.gguf -p "What is machine learning?"
Output:
llm_load_tensors: ggml ctx size = 0.14 MiB
llm_load_tensors: CPU buffer size = 3917.87 MiB
...
...
...
llama_print_timings: load time = 1602.50 ms
llama_print_timings: sample time = 11.38 ms / 571 runs ( 0.02 ms per token, 50193.39 tokens per second)
llama_print_timings: prompt eval time = 81.22 ms / 6 tokens ( 13.54 ms per token, 73.88 tokens per second)
llama_print_timings: eval time = 24270.78 ms / 570 runs ( 42.58 ms per token, 23.49 tokens per second)
llama_print_timings: total time = 24522.14 ms / 576 tokens
Build from source for GPU acceleration with ROCm:
git clone https://github.com/ggerganov/llama.cpp
cd llama.cpp
make GGML_HIPBLAS=1 AMDGPU_TARGETS=gfx1100 -j$(lscpu | grep '^Core(s)' | awk '{print $NF}')
Remarks: Use GGML_HIPBLAS
instead of LLAMA_HIPBLAS
To run:
./llama-cli -t $(lscpu | grep '^Core(s)' | awk '{print $NF}') --temp 0 -m ../mistral.gguf -p "What is machine learning?" -ngl 33
Output:
ggml_cuda_init: found 2 ROCm devices:
Device 0: Radeon RX 7900 XTX, compute capability 11.0, VMM: no
Device 1: Radeon RX 7900 XTX, compute capability 11.0, VMM: no
llm_load_tensors: ggml ctx size = 0.41 MiB
llm_load_tensors: offloading 32 repeating layers to GPU
llm_load_tensors: offloading non-repeating layers to GPU
llm_load_tensors: offloaded 33/33 layers to GPU
llm_load_tensors: ROCm0 buffer size = 1989.53 MiB
llm_load_tensors: ROCm1 buffer size = 1858.02 MiB
llm_load_tensors: CPU buffer size = 70.31 MiB
...
...
...
llama_print_timings: load time = 3440.52 ms
llama_print_timings: sample time = 17.78 ms / 952 runs ( 0.02 ms per token, 53540.30 tokens per second)
llama_print_timings: prompt eval time = 12.38 ms / 6 tokens ( 2.06 ms per token, 484.46 tokens per second)
llama_print_timings: eval time = 8928.70 ms / 951 runs ( 9.39 ms per token, 106.51 tokens per second)
llama_print_timings: total time = 9119.94 ms / 957 tokens
The difference in speed is more than obvious.
https://github.com/eliranwong/MultiAMDGPU_AIDev_Ubuntu/blob/main/benchmark.md
- Adjust the number of layers with
-ngl
to the maximum possible vaule for loading the files on your devices. - You may want to control the context size and the output tokens too.
For examples:
./llama-cli -m ../gguf/command-r-plus.gguf -p "What is machine learning?" --temp 0.0 -ngl 20 -c 2048 -n 2048 -t 24 -ngl 48
./llama-cli -m ../gguf/wizardlm2_8x22b.gguf -p "What is machine learning?" --temp 0.0 -ngl 20 -c 2048 -n 2048 -t 24 -ngl 34
The author managed to installed llama.cpp with
CMAKE_ARGS="-DLLAMA_CLBLAST=on" pip install llama-cpp-python
Alternately,
Use hipBLAS (ROCm) as backend:
sudo apt install libc6-dev libstdc++-12-dev
CMAKE_ARGS="-DLLAMA_HIPBLAS=on" pip install llama-cpp-python
Use Vulkan as backend:
CMAKE_ARGS="-DLLAMA_VULKAN=on" pip install llama-cpp-python
Read more at: https://llama-cpp-python.readthedocs.io/en/stable/
an issue regarding tensor_split feature
Edit the file "llama_cpp.py", in this case, located in '~/apps/freegenius/lib/python3.10/site-packages/llama_cpp'
Edit the file manually: '~/apps/freegenius/lib/python3.10/site-packages/llama_cpp/llama_cpp.py'
Change:
from
LLAMA_MAX_DEVICES = _lib.llama_max_devices()
to
#LLAMA_MAX_DEVICES = _lib.llama_max_devices()
LLAMA_MAX_DEVICES = 2
https://github.com/eliranwong/freegenius/wiki/Llama.cpp-Server-with-GPU-Acceleration
Setup of stable-diffusion-webui is straightforward as follows:
sudo apt install google-perftools libgl1
git clone https://github.com/AUTOMATIC1111/stable-diffusion-webui
cd stable-diffusion-webui
python3 -m venv venv
./webui.sh
# Set up an alias [optional]
echo 'alias sdwebui="'$(pwd)'/webui.sh"' >> ~/.bashrc
open http://127.0.0.1:7860
Read more at: https://github.com/AUTOMATIC1111/stable-diffusion-webui/wiki/Install-and-Run-on-AMD-GPUs
Official setup instructions works:
git clone https://github.com/comfyanonymous/ComfyUI
cd ComfyUI
python3 -m venv venv
source venv/bin/activate
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm6.1
pip install -r requirements.txt
python3 main.py
# Set up an alias [optional]
echo 'alias comfyui="'$(pwd)'/venv/bin/python3 '$(pwd)'/main.py"' >> ~/.bashrc
open http://127.0.0.1:8188
Install pipx xsel and ffmpeg to work with fabric:
sudo apt install -y pipx xsel ffmpeg
git clone https://github.com/danielmiessler/fabric.git
cd fabric
pipx install .
tee --append $HOME/.bashrc <<EOF
alias pbcopy='xsel -b -i'
alias pbpaste='xsel -b -o'
EOF
fabric --setup
source $HOME/.bashrc
Install docker first and run:
sudo apt install -y git
git clone https://github.com/ItzCrazyKns/Perplexica.git
cd Perplexica
cp sample.config.toml config.toml
docker compose up -d
open localhost:3000
For performance optimization, you may read:
https://huggingface.co/docs/optimum/main/en/amd/amdgpu/overview
https://github.com/nktice/AMD-AI/blob/main/performance-tuning.md
https://keras.io/guides/distributed_training_with_jax/
https://github.com/vosen/ZLUDA
Current known issues of ZLUDA: https://github.com/vosen/ZLUDA#known-issues