@owlairaposted at 2024-08-15

Radeon GPUを活用して、PytorchやKrasのパッケージを使いたいです

Q&A

Closed

Keras PyTorch ROCm Windows11 rx6950

自作PCのリソースを活用して、Deep Learningを実施したいです

以前AIの勉強をした時になじみのある、Kras・Pytorchを使いたいと思っています。
しかし、PythonでGPUを認識させることができず困っています。
有識者の方、アドバイスをお願いいたします。

課題

AMDのドライバーインストール→ドライバに対応したpytorchのインストール
上記の手順がうまくいきません。

以下の構成で自作PCを組んでいます

ハード構成

　・CPU：AMD Ryzen 9 5900X BOX
　・マザーボード：ASRock B550 Taichi
　・GPU：MSI Radeon RX 6900 XT GAMING Z TRIO 16G [PCIExp 16GB]

システム構成

　・Windows 11
　・64 ビットオペレーションシステム
　・×64 ベースプロセッサ
　・23H2
　・Ubuntu 22.02.3 (WSL)

発生している問題・エラー

hsa api call failure at: /long_pathname_so_that_rpms_can_package_the_debug_info/src/rocminfo/rocminfo.cc:1306
Call returned HSA_STATUS_ERROR_OUT_OF_RESOURCES: The runtime failed to allocate the necessary resources. This error may also occur when the core runtime library needs to spawn threads or create internal OS-specific events

Ubuntu

rocminfo #ドライバーがインストールされていることを確認するコマンド

自分で試したこと

　・GUIでのドライバーインストール
　・WSLを使った、ROCmのインストール。

AMDの公開しているドキュメントの方法は一通り試したつもりです。
WSL上での環境構築であることが依存しているかもしれないと考えております。
今後はLinuxをデュアルブートすることも検討しています。

1Answer

@megchandesu posted at 2024-08-15

AMDのドライバーインストール→ドライバに対応したpytorchのインストール
上記の手順がうまくいきません。

上記手順のどこが上手く行っていないのでしょうか？
また、OSやPythonのバージョン等Pytorchインストールの要件は満たされていますか？

Start Locally

0Like

Comments

@owlaira
Questioner
コメントありがとうございます。
説明不足で申し訳ありません。
OSはWindows11でpython3.8をanacondaで使っています。
正規の手順は
①GPU対応のドライバー（CUDA、ROCｍ）のインストール
②ドライバーに対応した方法で、pytotchをpython環境にインストール
③プロンプト上でのGPUの確認
Python
import torch print(torch.cuda.is_available())
output
True
しかし、下記の写真のように、Windows OSでROCmに対応したPytorchは、ダウンロードオプションが存在しない状態です。（ご提示いただいたダウンロードリンクです）
(https://qiita-image-store.s3.ap-northeast-1.amazonaws.com/0/3618474/91dbd2f0-7591-2d17-6899-310e0fde0621.png)
そこで　Windows subsystem for Linux（WSL）を活用して、ドライバーダウンロードを試みました。手順①に該当します。
ドライバーの動作を確認するために、WSL上でGPUの認識を確認するコマンドがあるのですが、その過程でエラーが出てしまうといった状態です。
https://rocm.docs.amd.com/projects/radeon/en/latest/docs/install/wsl/install-radeon.html

手順①がうまくいっていない状態です。
@megchandesu
・提示いただいたリンクのPrerequisitesは満たされている状態でしょうか？
・ドライバーインストール中に何かエラーや警告は出ませんでしたか？
@himura57
https://rocm.docs.amd.com/projects/radeon/en/latest/index.html
現行の6.1.3の動作要件がRDNA3以降(RX7000シリーズ)になっているのでそこで弾かれているのではないでしょうか
@owlaira
Questioner
解決しました。ありがとうございます。
お二方のご指摘いただきましたように、現環境は要件を満たせていないようです。
具体的には、windows11、且つ、RDNA2(RX 6950)の環境ですと、rocmがサポートしている要件を満たせないことがわかりました。

Linux環境で動かす方向にシフトしたいと思いますが、WSLだとうまくいかない等の報告も上がっているため、ubuntu 22.04でデュアルブートしました。
Linuxでしたら、amdGPU経由でROCm6.1をインストールして、Pytorchも問題なくインストールできました。

Are you sure you want to delete the question?