一、下载
https://pytorch.org/
下载release版本并解压(参考网上博客)
二、vs2017配置
-
右击解决方案-->属性-->C/C++-->常规-->附加包含目录
如图,添加libtorch下的两个文件夹
-
右击解决方案-->属性-->链接器-->常规-->附加目录项
添加lib路径
-
右击解决方案-->属性-->链接器-->输入-->附加依赖项
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,则环境配置成功
四、新坑记录
符合模式选否(默认是)