天天看點

Libtorch + vs 2019安裝及配置

Libtorch是c++接口的Pytorch,提供了與python版本基本一緻的接口,使用很友善,而且更适合實際部署。這裡介紹Libtorch的在VS2019下的部署和簡單測試使用。

1、從官網下載下傳

https://pytorch.org/get-started/locally/

目前可以下載下傳到Libtorch1.5版本,如果電腦支援Gpu,且安裝了Cuda,可以下載下傳Gpu版本,否則下載下傳cpu版本。

Libtorch + vs 2019安裝及配置

下載下傳解壓之後如下

Libtorch + vs 2019安裝及配置

2、在VS2019中的配置

需要配置.頭檔案h 庫檔案,lib, 和dll檔案

配置頭檔案

Libtorch + vs 2019安裝及配置

配置依賴庫的位置

Libtorch + vs 2019安裝及配置

添加依賴庫的檔案名

Libtorch + vs 2019安裝及配置

設定dll檔案的環境變量,

PATH=D:\software\gnsoftware\pyTorch\libtorch-win-shared-with-deps-debug-1.5.0\libtorch\lib;%PATH%

(之前我一般是把那一堆dll檔案拷貝到生成的exe檔案所在目錄,導緻這些dll檔案會複制很多份,占據很大空間,這裡通過設定依賴于項目的環境變量,少了大量拷貝,還不影響其他變量)

Libtorch + vs 2019安裝及配置

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

Libtorch + vs 2019安裝及配置

繼續閱讀