AMD GPUにおけるROCmを用いた深層学習環境の構築と検証

概要AI分野においてNVIDIAのCUDAエコシステムが広く普及している一方で、AMDはROCm (Radeon Open Compute) を通じて対抗している。ROCmは、AMDのGPUアクセラレーションおよびヘテロジニアスコンピューティングを支援するための開発プラットフォームである。本稿では、AMDの統合型GPU(APU)環境上にROCmを構築し、PyTorchやOpenCLを用いた動作検証を行うとともに、その裏側 ...

6月29日 22:55 投稿