当前位置: 代码迷 >> 综合 >> Win10+libtorch+CUDA+vs2017
  详细解决方案

Win10+libtorch+CUDA+vs2017

热度:101   发布时间:2023-11-21 01:41:54.0

在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!拜拜
在这里插入图片描述