Libtorch是c++接口的Pytorch,提供了與python版本基本一緻的接口,使用很友善,而且更适合實際部署。這裡介紹Libtorch的在VS2019下的部署和簡單測試使用。
1、從官網下載下傳
https://pytorch.org/get-started/locally/
目前可以下載下傳到Libtorch1.5版本,如果電腦支援Gpu,且安裝了Cuda,可以下載下傳Gpu版本,否則下載下傳cpu版本。
下載下傳解壓之後如下
2、在VS2019中的配置
需要配置.頭檔案h 庫檔案,lib, 和dll檔案
配置頭檔案
配置依賴庫的位置
添加依賴庫的檔案名
設定dll檔案的環境變量,
PATH=D:\software\gnsoftware\pyTorch\libtorch-win-shared-with-deps-debug-1.5.0\libtorch\lib;%PATH%
(之前我一般是把那一堆dll檔案拷貝到生成的exe檔案所在目錄,導緻這些dll檔案會複制很多份,占據很大空間,這裡通過設定依賴于項目的環境變量,少了大量拷貝,還不影響其他變量)
4、簡單的測試例子
#include <iostream>
#include <torch/torch.h>
int main()
{
torch::Tensor tensor = torch::rand({ 5,3 });
std::cout << tensor << std::endl;
return EXIT_SUCCESS;
}
運作結果如下,說明配置沒問題,可以開始後面的Libtorch學習之旅。