ROCm

ROCm3.5でPyTorchビルド

前回の続き?のような何か。 maminus.hatenadiary.org 今回やったこと ROCm3.5 Docker環境 PyTorchをmasterブランチでビルド torchvisionをmasterブランチでビルド 上記に伴い公式にバグ報告 前回同様PyTorch版EfficientDet1で動作確認 前回からの変化点を中…

PyTorch版EfficientDetをROCm環境で実行してみた

PyTorch版EfficientDet1のカメラで撮影した画像に推論結果を重畳してくれるデモをROCm環境で動かしたので手順などをメモ。 環境 Radeon RX 580 Ubuntu 18.04.4 LTS(ホスト側) dockerイメージ Docker Hubでrocm/pytorchの中からそれっぽいのを選択。 Ubuntu…