背景
ROCm(AMD GPU)もそれなりに安定して動くようになった. けどまだまだ CUDA も使いたいときや, ROCm での動作がおかしいときに CUDA で実行確認したいときがある.
構成
Ubuntu 20.04 or 22.04 とします.
以前だとなんかうまく同時認識されなかったりしましたが,
ROCm 5.2 以降あたりだと, 特には問題なく動くような気はしています.
- Intel z590 マザー
- AMD X570 マザー
の両方で動作確認しました. より安定を求めるなら x299, x399 あたりのワークステーション系がいいかもしれません
ROCm pytorch は Docker で
pytorch で両 GPU を同時に使うのは無理でしょう.
conda で環境切り替えでの対応でもよいかもですが,
でのインストールでは ROCm バージョン古いままなので, ROCm のほうは Docker で運用がよいと思われます.