libtorch-GPU在vs2019上的配置

libtorch-GPU在vs2019上的配置

vs2019属性表的配置
包含目录：

    D:\VS2017Library\libtorch\include

    D:\VS2017Library\libtorch\include\torch\csrc\api\include

库目录：

  D:\VS2017Library\libtorch\lib

链接器附加依赖项：

    torch.lib
    torch_cuda.lib
    torch_cuda_cu.lib
    torch_cuda_cpp.lib
    torch_cpu.lib
    c10_cuda.lib
    caffe2_nvrtc.lib
    c10.lib
    kineto.lib
    dnnl.lib
    fbgemm.lib
    asmjit.lib
    XNNPACK.lib
    cpuinfo.lib
    clog.lib
    libprotoc.lib
    pthreadpool.lib
    libprotobuf.lib
    libprotobuf-lite.lib

最后，还有一个非常重要的设置要做，这就是GPU版跟CPU版配置最大的两个区别之一，在我们的vs2017找到项目>>>Project3属性>>>链接器>>>命令行，输入：

/INCLUDE:“?ignore_this_library_placeholder@@YAHXZ”

测试代码：

#include <torch/torch.h>
#include <torch/script.h>

using namespace std;

int main()
{
torch::DeviceType device_type = at::kCPU;
if (torch::cuda::is_available()) {
cout << “cuda可以使用” << endl;
torch::DeviceType device_type = at::kCUDA;
}
else
{
cout << “cpu” << endl;
}

}

libtorch教程：https://zhuanlan.zhihu.com/p/369930315