如何#include<torch.extension.h> 在 Visual Studio 的 .cpp 文件中？

Question

現在我正在處理一個需要使用 Pytorch C++ 擴展的項目。 我已經在 python 虛擬環境中安裝了 1.4.0 版本的 Pytorch：

activate crfasrnn
>>>import torch
>>>print(torch.__version__)
1.4.0

我使用 Visual Studio 作為 C++ 編譯器。 代碼是這樣的：

#include<iostream>
#include<torch/extension.h>

int main()
{
    std::cout << "hello, world!" << std::endl;
    return 0;
}

但是，編譯器告訴我“無法打開'torch/extension.h'”我該如何解決這個問題？

Answer 1

您需要從https://pytorch.org/獲取適當的 LibTorch 二進制文件，然后在 Visual Studio 中包含所需的二進制文件。 您可以按照此處給出的詳細說明進行操作。