Ryzen7 5700Uなど、CPU内蔵のGPUで、PyTorchを利用する
amdのcudaに相当するrocmのinstall方法は、対応のドライバーのインストールとruntime環境のinstallからなります
# Make the directory if it doesn't exist yet.
# This location is recommended by the distribution maintainers.
sudo mkdir --parents --mode=0755 /etc/apt/keyrings
# Download the key, convert the signing-key to a full
# keyring required by apt and store in the keyring directory
wget https://repo.radeon.com/rocm/rocm.gpg.key -O - | \
gpg --dearmor | sudo tee /etc/apt/keyrings/rocm.gpg > /dev/null
echo "deb [arch=amd64 signed-by=/etc/apt/keyrings/rocm.gpg] https://repo.radeon.com/amdgpu/6.1/ubuntu jammy main" \
| sudo tee /etc/apt/sources.list.d/amdgpu.list
sudo apt update
sudo apt install amdgpu-dkms
sudo reboot
sudo apt remove rocminfo rocm-smi rocm-device-libs
sudo apt install rocm
sudo tee --append /etc/ld.so.conf.d/rocm.conf <<EOF
/opt/rocm/lib
/opt/rocm/lib64
EOF
sudo ldconfig
.bashrc
export PATH=$PATH:/opt/rocm-6.1.0/bin
export HSA_OVERRIDE_GFX_VERSION=9.0.0
export HSA_ENABLE_SDMA=0
を追加
check
dkms status
rocminfo
clinfo
reinstall pytorch
python3 -m pip uninstall torch torchvision torchaudio
https://pytorch.org/get-started/locally/
を参考に
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
trouble shoot
もし、rocminfoが、通常ユーザーで動かなかったら
$ ls -la /dev/kfd
crw-rw---- 1 root render 235, 0 5月 8 15:04 /dev/kfd
上は動いているときの状況です。
sudo usermod -aG render $USER
sudo usermod -aG video $USER
や、
sudo chmod g+rw /dev/kfd
などパーミッションの設定を