背景
在vs2019 c++项目中使用libtorch,为加快算法速度,将算法中模型和输入放入GPU进行运算。
问题描述
使用torch::cuda::is_available(),检查gpu是否可用时,下方出现红色波浪线,无法完成编译。
解决方法
在程序顶部添加头文件
#include <torch/torch.h>
错误即可解决!!
tips:
其它CUDA相关函数:
//cudnn是否可用
std::cout << "torch::cuda::cudnn_is_available:" << torch::cuda::cudnn_is_available() << endl;
//可用的GPU个数
std::cout << "torch::cuda::device_count:" << torch::cuda::device_count() << endl;
-------------------------------end-------------------------------------
更多推荐
【常见问题】torch::cuda::is_available()语句报错
发布评论