我现在正在处理一个需要使用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’”。我该如何解决这个问题?
回答:
你需要从https://pytorch.org/获取适当的LibTorch二进制文件,然后在Visual Studio中包含所需的二进制文件。你可以按照这里提供的详细说明进行操作。