Ryzen で機械学習 - インゲージ開発者ブログ

こんにちは、masm11 です。 私は以前、機械学習を勉強したことがありまして、 CPU での機械学習を実装したり NVIDIA GPU で実装したりは経験があったのですが、 Ryzen APU での実装は経験がなく、せっかく手元に実機があるので、やってみました。 ハードウェア環境 PC は HP ENVY x360 13 (ay0000) です。 この PC には AMD Ryzen 7 4700U という APU が搭載されています。 今回は、この 13.3" というコンパクトなノート PC で PyTorch で MNIST (手書き数字の判別) をやります。 コンテナを用意して起動する docker run -it --network=host --device=/dev/kfd --device=/dev/dri --group-add=video --ipc=host --ca