タグ

ブックマーク / zenn.dev/hidetoshi (1)

  • PyTorchをM1 MacBook のGPU(MPS)で動かす.実行時間の検証もしたよ

    最近,PyTorchがM1 MacBookGPUに対応したとのことで,そのインストール方法を説明します.また,簡単に計算時間を検証してみたので,その結果を共有します. この記事は,ある程度PyTorchを使い慣れている人向けの記事です. さっくりとM1 MacBookGPUを使って深層学習(ディープラーニング)を動かしてみたい人にちょうど良いかと思います. M1 MacBookGPUを使うためのPyTorchのインストール方法 (2023/03/30 更新) ※ 記事初公開時はStable版がなかったのですが,2023/03/30時点でStable版(ver2.0.0)があったのでインストール方法を修正しています. 筆者はインストールのために以下のページを参考にしました. pipもしくはcondaでStable版のインストールが可能です. 以下のいずれかのコマンドを実行することで,

    PyTorchをM1 MacBook のGPU(MPS)で動かす.実行時間の検証もしたよ
    CLSmooth
    CLSmooth 2022/09/04
  • 1