背景

在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-------------------------------------