Library not loaded: libmklml.dylib use c++ front

Mac OSでPyTorchのC++ APIのコードを実行しようとすると以下のエラーが出る． dyld: Library not loaded: @rpath/libmklml.dylib Referenced from: /Users/hzq/Documents/www/test/libtorch/lib/libcaffe2.dylib Reason: image not found Abort trap: 6 libm…

2019-01-14

jupyter notebookでPyTorch C++ APIを試してみる

C++ PyTorch PyTorch C++ API

jupyter notebookでC++を動かすカーネルであるxeus-clingとPyTorchのC++ APIであるlibtorchを利用して，jupyter notebook上でPyTorchをC++から使ってみる．準備 LibTorchのインストールはじめに，PyTorchのC++ APIであるLibTorchをインストールしておく． …

2019-01-11

PyTorch C++ APIでGPUを使ってみる

C++ PyTorch PyTorch C++ API

今回はPyTorchのC++ APIでGPUを使った学習と推論を試してみる． GPUの使い方 C++でGPUが使えるかどうかは，以下のように確認できる torch::cuda::is_available() ここはpythonとほとんど変わらない GPUが使えることを確認したら，モデルをGPUに転送する torc…

2019-01-10

PyTorch C+ APIで自作データセットを定義する．

PyTorch C++ API PyTorch C++

PyTorchのC++ APIで自作のデータセットの定義の仕方を紹介．今回は以下のようなデータセットを作る，データセットクラス：ImageDataset(root, labelfile) 画像が格納されたルートディレクトリ：root 画像は以下のように格納 $ tree test_data/ test_data/ …

2019-01-08

PyTorchのC++ APIを試してみる

C++ PyTorch PyTorch C++ API

PyTorchではC++ APIが提供されている． pytorch.org 多層パーセプトロン版ビルド $ mkdir build $ cd build $ cmake .. $ cmake -DCMAKE_PREFIX_PATH=/absolute/path/to/libtorch .. $ make #include <torch/torch.h> struct Net : torch::nn::Module { Net() { fc1 = regi</torch/torch.h>…