Ryzen5 5600GでROCmを使いstable-diffusion-webuiを試した話

コンピュータ

AI・機械学習

2024.02.052024.04.01

調べたところCUDAの代わりにROCmという物があって、こちらに対応したPyTorchがあるのでそれを使えばAMDのGPUでstable-diffusion-webuiが実行することが出来るらしいです。

https://pytorch.org/

ただしグラフィックボードに搭載されたGPU向けで、Ryzen5 5600GなどのCPU内蔵のGPUは非対応とのことです。

私が所有するRyzen5 5600Gでは無理ということに成りますが、環境変数設定でGPUの名前を対応する物に変更すれば実行することが出来る情報がありましたので、試してみました。

Linux ROCm経由でのRyzen 5600G GPUによるStable Diffusion WebUIの利用 - Qiita

画像生成AI Stable Diffusion画像生成AIのStable Diffusionの統合環境であるStable Diffusion WebUI(AUTOMATIC1111版)を、AMDの…

環境構築を行い実行してみたところ動くところまでは確認できました。512×512の画像を１枚作成するのに1分ぐらいかかる感じです。ちなみにDeskMiniX300にDDR4メモリを物理的に32GBつみ、VRAMに16GB割り当てしています。VRAMが16GBのグラフィックボードは結構な値段がしますが、こちらのメモリは32GBで1万円しません。とりあえずVRAMの容量だけは立派なマシンが安価に構築出来ます。これだけVRAMがあれば動作は遅いですが、動作はしてくれます。

最近発売された8000Gシリーズは結構な処理能力があるらしく高速なDDR5メモリとの組み合わせになりますので、こちらで動作させることが出来れば、少し面白いかもしれません。