TamaGo icon indicating copy to clipboard operation
TamaGo copied to clipboard

Add support for apple silicon

Open zakki opened this issue 9 months ago • 4 comments

Macでuse-gpuを指定した時にmpsバックエンドを使うようにします。

M3 Ultra環境だとデフォルトネットワークで16ワーカー32visitsで22kgames/hourくらい自己対局出来ています。

zakki avatar Apr 02 '25 11:04 zakki

学習側は2.6.0では未対応の処理があってワークアラウンドが必要でした。

https://github.com/pytorch/pytorch/issues/142397

zakki avatar Apr 08 '25 13:04 zakki

requirements.txtのtorchのバージョンを合わせておいていただけますか。

kobanium avatar May 08 '25 14:05 kobanium

PyTorchのmainブランチでは修正されていますが、その後リリースされた 2.7.0 や 2.7.1-rc1 は別ブランチでの開発らしく反映されていません。 マージはPyTorch 2.8以降がリリースされてからのほうが良いかもしれません。

ちなみに修正はこれです。 https://discuss.pytorch.org/t/gradscaler-typeerror-cannot-convert-a-mps-tensor-to-float64-dtype-as-the-mps-framework-doesnt-support-float64-please-use-float32-instead/213206

zakki avatar May 09 '25 04:05 zakki

ありがとうございます。問題点を理解できました。 PyTorchの修正が済んでからマージします。

kobanium avatar May 09 '25 06:05 kobanium