背景

ROCm(AMD GPU)もそれなりに安定して動くようになった. けどまだまだ CUDA も使いたいときや, ROCm での動作がおかしいときに CUDA で実行確認したいときがある.

構成

Ubuntu 20.04 or 22.04 とします.

以前だとなんかうまく同時認識されなかったりしましたが,
ROCm 5.2 以降あたりだと, 特には問題なく動くような気はしています.

の両方で動作確認しました. より安定を求めるなら x299, x399 あたりのワークステーション系がいいかもしれません

pytorch で両 GPU を同時に使うのは無理でしょう.

conda で環境切り替えでの対応でもよいかもですが,

でのインストールでは ROCm バージョン古いままなので, ROCm のほうは Docker で運用がよいと思われます.