在VS中配置libtorch,之前下载过libtorch的多个版本,在VS2015和VS2013上都配置失败,换成VS2017后成功了。
1、到官网下载相应版本的libtorch
本人下载的版本
libtorch1.7 - cuda10.2
2、解压下载的libtorch压缩包
解压后看到如下的文件夹,很标准,那就安装opencv的方式来配置它
3、开始搞它
完全安装配置opencv的方式来盘它
3.1 include
把include文件夹弄进去
3.2 lib路径
3.3 lib
把lib文件夹下的所有.lib给它放进去,因为我不知道要用到哪个
4、水来土掩
编译,结果,哎呀呀,缺少各种.dll,在lib下明明有一堆dll,怎么说找不到呢?
解决:
给你一个调试时的搜索路径吧,就是dll所在的路径,注意修改哦
PATH=D:\softwarePackage\torch\libtorch_debug_171_gpu\lib;%PATH%
这次编译成功了!
5、结束?
结束是不可能的
#include<torch/torch.h>
#include<torch/script.h>
#include<iostream>int mian(){
std::cout << "cuda::is_available():" << torch::cuda::is_available() << std::endl;
std::cout << "torch::cuda::cudnn_is_available():" << torch::cuda::cudnn_is_available() << std::endl;
std::cout << "torch::cuda::device_count():" << torch::cuda::device_count() << std::endl;
}
结果:找不到可用的GPU,不可能呀
百度朋友给的答案是:
/INCLUDE:?warp_size@cuda@at@@YAHXZ
6、ok!拜拜