Libtorch + vs 2019安裝及配置

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

 

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章