libtorch gpu vs2017环境配置

一、下载

https://pytorch.org/
下载release版本并解压（参考网上博客）

二、vs2017配置

1. 右击解决方案-->属性-->C/C++-->常规-->附加包含目录
   

   如图，添加libtorch下的两个文件夹

2. 右击解决方案-->属性-->链接器-->常规-->附加目录项
   

   添加lib路径

3. 右击解决方案-->属性-->链接器-->输入-->附加依赖项

   
   

   c10.lib
   c10_cuda.lib
   torch_cpu.lib
   torch_cuda.lib
   torch_cuda_cu.lib
   torch_cuda_cpp.lib
   -INCLUDE:?warp_size@cuda@at@@YAHXZ
   最后一项十分重要
   4.右击解决方案-->属性-->链接器-->输入-->命令行

   
   

三、测试

#include<iostream>
#include<torch/script.h>
#include <torch/torch.h> 
#include<memory>
 
int main()
{
    std::cout <<"cuda::is_available():" << torch::cuda::is_available() << std::endl;
    system("pause");
}

如果结果为1，则环境配置成功

四、新坑记录

符合模式选否（默认是）