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學習之旅。