环境说明:windows10,显卡3090,已安装cuda11.4,IDE为VS2019,已配置opencv4.5.5
主要参考:https://blog.csdn.net/qq_41562735/article/details/125922729
1、下载libtorch1.11.0的windows版本(Release版本):点击下载
2、解压到本地,设置在VS2019中添加属性表或直接设置项目属性:
3、如果报错找不到.dll,可以尝试的办法有:①从解压的libtorch包中找到这些dll的路径,将其添加到环境变量。②如果不行,将这些dll复制到C:\Windows\System32。③如果还是不行,将这些dll复制到当前项目的Release文件夹下。
4、运行测试代码
#include "torch/torch.h"
#include "torch/script.h"
void main()
{
std::cout << "这是生成张量操作!" << std::endl;
torch::Tensor output = torch::randn({ 3,2 });
std::cout << output << std::endl;
std::cout << "CUDA:" << torch::cuda::is_available() << std::endl;
std::cout << "CUDNN: " << torch::cuda::cudnn_is_available() << std::endl;
std::cout << "GPU(s): " << torch::cuda::device_count() << std::endl;
system("pause");
}
输出如下则表示配置成功
5、如果输出为都为0,在“属性 --> 链接器 --> 命令行 --> 其他选项”中添加:
/INCLUDE:"?ignore_this_library_placeholder@@YAHXZ"