A カード ユーザーは、絵を描いたり、エリクサーを作成したり、大規模なモデルを実行したり、さらには機械学習や深層学習を実行したりできれば幸運です~
注意:ROCm目前仅限在Linux系统下可用,Windows暂不支持
1.ROCmをインストールする
RX6000 シリーズ以下のグラフィックス カードは ROCm 5.4.2 安定版コマンドを使用します
[サポートにはデスクトップ AMD Radeon RX6950XT、RX6900XT、RX6800XT、RX6800、RX6750XT、RX6750GRE、RX6700XT、RX6700GRE、RX6700、RX6650XT、RX6600XT、RX6600、RX6500XT、RX6400 およびその他のモバイル AMD Radeon RX 6850M XT、RX が含まれます6800M、RX6700M、RX6650M XT、
RX6800S 、RX6700S、RX6650M、RX6600S、RX6600M、RX6500M、RX6300Mなど】
pip install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/rocm5.4.2
新しい RX7000 シリーズ グラフィックス カードには ROCm バージョン 5.6 (さらに、--pre
最新のプレリリース バージョンまたは開発バージョンを照会するオプション)が必要です。
[サポートにはデスクトップ AMD Radeon RX7900 XTX、RX7900 XT、RX7800 XT、RX7700 XT、Radeon RX7600 など、
モバイル RX 7600M XT などが含まれます。
pip install --pre torch torchvision torchaudio --index-url https://download.pytorch.org/whl/nightly/rocm5.6
2. インストールが完了したら、Python 環境で cuda が利用できるかどうかを確認します。
import torch
torch.cuda.is_available()
コードの実行結果の出力はtrue
OKです
3. プロジェクトを実行する前に、必要な環境変数の設定があります。
3.1 まず、Linux 構成ファイルを開きます。
vim ~/.bashrc
3.2 最後に次の行を追加します。
グラフィックス カードが RX6000 シリーズ以下の場合は、これをコピーして貼り付けます。
エクスポート HSA_OVERRIDE_GFX_VERSION=10.3.0
RX7000 シリーズのグラフィック カードの場合は、これをコピーして貼り付けます。
エクスポート HSA_OVERRIDE_GFX_VERSION=11.0.0
3.3 設定を保存して終了し、設定をすぐに有効にします。
source ~/.bashrc
全文