LibTorch入門——Win10+VS2017配置LibTorch開發(fā)環(huán)境。

最近想研究一下如何部署pytorch模型，看到別人說LibTorch不錯，是最近流行的C++部署框架，我打算試試。

首先，去pytorch官網(wǎng)上下載，如下圖，上面的連接是release版本，下面是Debug版本，支持C++，但是Window不支持Java。點擊連接下載下來。我選擇的是release版本的。

將解壓后的文件復(fù)制到C盤，如下圖：

設(shè)置環(huán)境變量：

右鍵我的電腦->高級系統(tǒng)設(shè)置->高級中的環(huán)境變量->點擊系統(tǒng)變量中的Path->添加dll路徑：

.

然后新建C++工程文件

新建完成后默認(rèn)是x86的，需要改為X64的，因為libtorch是x64的，不支持32位的。我下載是release版本的，所以還要改為release模式。

然后右鍵lib_torch項目,選擇屬性，配置libtorch環(huán)境。

選擇C\C++? ——》常規(guī)，將SDL檢查改為否。添加附加包含目錄。

改完后的效果如下圖：

將 C/C++ => 語言 => 符合模式，改為否。

設(shè)置連接庫，選擇連接器——》常規(guī)，修改附加庫目錄

設(shè)置連接庫，選擇連接器——》輸入，修改附加依賴項

在附件依賴項中增加：

c10.lib

mkldnn.lib

libprotobuf.lib

torch.lib

torch_cpu.lib

在 配置屬性à調(diào)試à環(huán)境 里添加libtorch動態(tài)庫的路徑

PATH=C:\libtorch\lib;%PATH%

到這里就配置完了，寫入測試代碼試一下。

#include

#include "torch/script.h"

#include

int main()

{

torch::Tensor output = torch::randn({ 5,2 });

std::cout << output;

return 0;

}

出現(xiàn)下面的結(jié)果說明配置成功了。

C++

版權(quán)聲明：本文內(nèi)容由網(wǎng)絡(luò)用戶投稿，版權(quán)歸原作者所有，本站不擁有其著作權(quán)，亦不承擔(dān)相應(yīng)法律責(zé)任。如果您發(fā)現(xiàn)本站中有涉嫌抄襲或描述失實的內(nèi)容，請聯(lián)系我們jiasou666@gmail.com 處理，核實后本網(wǎng)站將在24小時內(nèi)刪除侵權(quán)內(nèi)容。