ubuntuにROCm 6.xを導入

前にも同じような記事を書いたんだけど、導入が簡単になってたのでちょっと書いてみる。
参考先URL:Release notes for AMD ROCm™ 6.0
sudo apt install rocm
以上。
超簡単になってた。
renderとかvideoの設定はしないとダメなのかまでは検証していないので、そのへんは過去記事を見てみてください。

staible diffusionでROCm6.xを有効にするには、venvディレクトリーを削除して、
pyton -m venv venv
source ./venv/bin/activate
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/rocm6.0
上記コマンドを入力。.webui.shを実行すれば、勝手にインストールしてくれるので以降はROCm 6.xを使用して画像生成できるようになります。

以前、ROCm 6.xを入れた時は、またpytorchがnightlyバージョンだった為かうまく動かなかったけど、現在は大丈夫です。
導入されているROCmのバージョンを確認するコマンドは、以下の通りです。
apt show rocm-libs -a

実行結果

Package: rocm-libs
Version: 6.0.2.60002-115~20.04
Priority: optional
Section: devel
Maintainer: ROCm Dev Support 
Installed-Size: 13.3 kB
Depends: hipblas (= 2.0.0.60002-115~20.04), hipblaslt (= 0.6.0.60002-115~20.04), hipfft (= 1.0.13.60002-115~20.04), hipsolver (= 2.0.0.60002-115~20.04), hipsparse (= 3.0.0.60002-115~20.04), hiptensor (= 1.1.0.60002-115~20.04), miopen-hip (= 3.00.0.60002-115~20.04), half (= 1.12.0.60002-115~20.04), rccl (= 2.18.3.60002-115~20.04), rocalution (= 3.0.3.60002-115~20.04), rocblas (= 4.0.0.60002-115~20.04), rocfft (= 1.0.25.60002-115~20.04), rocrand (= 3.0.0.60002-115~20.04), hiprand (= 2.10.16.60002-115~20.04), rocsolver (= 3.24.0.60002-115~20.04), rocsparse (= 3.0.2.60002-115~20.04), rocm-core (= 6.0.2.60002-115~20.04), composablekernel-dev (= 1.1.0.60002-115~20.04), hipblas-dev (= 2.0.0.60002-115~20.04), hipblaslt-dev (= 0.6.0.60002-115~20.04), hipcub-dev (= 3.0.0.60002-115~20.04), hipfft-dev (= 1.0.13.60002-115~20.04), hipsolver-dev (= 2.0.0.60002-115~20.04), hipsparse-dev (= 3.0.0.60002-115~20.04), hiptensor-dev (= 1.1.0.60002-115~20.04), miopen-hip-dev (= 3.00.0.60002-115~20.04), rccl-dev (= 2.18.3.60002-115~20.04), rocalution-dev (= 3.0.3.60002-115~20.04), rocblas-dev (= 4.0.0.60002-115~20.04), rocfft-dev (= 1.0.25.60002-115~20.04), rocprim-dev (= 3.0.0.60002-115~20.04), rocrand-dev (= 3.0.0.60002-115~20.04), hiprand-dev (= 2.10.16.60002-115~20.04), rocsolver-dev (= 3.24.0.60002-115~20.04), rocsparse-dev (= 3.0.2.60002-115~20.04), rocthrust-dev (= 3.0.0.60002-115~20.04), rocwmma-dev (= 1.3.0.60002-115~20.04)
Homepage: https://github.com/RadeonOpenCompute/ROCm
Download-Size: 1,044 B
APT-Sources: https://repo.radeon.com/rocm/apt/6.0.2 focal/main amd64 Packages
Description: Radeon Open Compute (ROCm) Runtime software stack