AMD GPU ROCm
Jump to navigation
Jump to search
AMD GPU ROCm
how to use amd grafic cards for speeding up calculations un tumbleweed similar tu nvidia with cuda
== Install rocm meta package
sudo zypper ar https://download.opensuse.org/repositories/science:/GPU:/ROCm:/Work/openSUSE_Tumbleweed/ science_GPU_ROCm sudo zypper ref sudo zypper in rocm sudo usermod -aG render # restart session
Test Installation
clinfo -l
Should output something like this:
Platform #0: AMD Accelerated Parallel Processing +-- Device #0: gfx1201 `-- Device #1: gfx1036
rocminfo | grep gfx
Should output something like
Name: gfx1201 Name: amdgcn-amd-amdhsa--gfx1201 Name: amdgcn-amd-amdhsa--gfx12-generic Name: gfx1036 Name: amdgcn-amd-amdhsa--gfx1036 Name: amdgcn-amd-amdhsa--gfx10-3-generic
PyTorch
for python projects like wyoming using pytorch add rocm support to a virtual environment. See https://pytorch.org/get-started/locally/ for the current final pip command
cd your/venv/basedirectory python3 -m venv wyoming . wyoming/bin/activate pip install --upgrade pip pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.4
Test with (should print True)
python -c 'import torch; print(torch.cuda.is_available())'