c 呼叫pytorch的模型遇到的問題

2021-09-11 02:42:55 字數 495 閱讀 1385

最近學習基於pytorch框架下的行為識別演算法(eco**),要將訓練好的模型在c++工程中用起來,需要走通c++呼叫pytorch的模型的路。在參考了官網的說明和網上已有的部落格,記錄下遇到的問題。

1.網上別人已經寫好的**github和教程

2.官網的介面說明

遇到的問題:

cmake時需要指明libtorch的路徑且需要是絕對路徑(在windows下,需要加上-a x64來指定64位,並且要在release下才能正確的匯入模型)

cmake -dcmake_prefix_path=path/to/libtorch ..
gcc編譯器版本需要5.3.1不然會報錯:

terminate called after throwing an instance of 'std::bad_alloc'  

what(): std::bad_alloc

aborted

pytorch的模型解析

如何獲取pytorch的動態圖?model torch.jit.load test.pth graph model.graph.copy torch.c.jit pass inline graph node list graph.nodes 載入模型後,獲取模型的graph,這個graph就是需要的...

安裝pytorch遇到的問題

git clone recursive pip install r requirements.txt 這步操作超級慢 python setup.py install 安裝成功後,測試 python import torch 出現錯誤 參考 cd usr local lib python3.5 dis...

pytorch 學習遇到的 問題

relu 和 leaky relu 中 inplace 設定 nn.relu inplace true default inplace false nn.leakyrelu inplace true default inplace false 將計算得到的值直接覆蓋之前的值 有時能夠節省運算記憶體,...