解决LibTorch中 Torch::cuda::is_available()返回为False和module.to(device)异常

1.Windows10系统下使用LibTorch 1.6.0,使用VS2017进行C++开发时,
Torch::cuda::is_available()返回值为0(False),显卡驱动也进行更新。

通过python import torch时,返回为True。所以cuda、cudnn和torch都没问题。

 最后找到这个方法:亲测可用。

在链接器中 -> 命令行 -> 其他选项

/INCLUDE:?warp_size@cuda@at@@YAHXZ

 再重新编译,运行Torch::cuda::is_available()返回值为True

2.Windows10系统下使用LibTorch 1.9.0,使用VS2019进行C++开发时,出现module.to(device)异常,如下图。

在国外论坛找到,还是在链接器中 -> 命令行 -> 其他选项,参数变化,需改为

/INCLUDE:?searchsorted_cuda@native@at@@YA?AVTensor@2@AEBV32@0_N1@Z

 然后就没有出现异常,可以GPU推理

 参考:解决LibTorch1.5.0 中 Torch::cuda::is_available()返回为False_耳东小白的小笔记本儿-CSDN博客

评论 10
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值