c++ 调用libtorch1.8GPU + cuda11.1+rtx3090+18.04 torch::cuda::is_available()为false 问题记录
1、使用cmake搭建工程:是网上常见的cmake建立的工程,这个是可以正常调用GPU在文件夹下建立三个文件CMakeLists.txt、example-app.cpp、buildCMakeLists.txt:cmake_minimum_required(VERSION 3.0 FATAL_ERROR)project(example-app)set(Torch_DIR /opt/libs-x64/libtorch_gpu1.8/share/cmake/Torch)find_packa..
原创
2023-01-13 12:39:52 ·
713 阅读 ·
0 评论