とある社会人ドクターのブログ

研究や読んだ書籍等の話をします

PytorchC++APIからJIT compileした訓練済みモデルを使う

概要

PytorchモデルをJITコンパイルしてWindowアプリにしたい.

libtorchの準備

libtorchをWIndowsでビルドする必要があります.こちらのリンクがそのまま使えると思います.

tadaoyamaoka.hatenablog.com

訓練済みモデルのコンパイル

pytorchのモデルをJITでコンパイル

C++からコンパイルしたモデルを呼びだす

Pytorch C++ API のやつ

Pytorch C++ APIのやつ

Pytorch C++ APIのメイン