CLion配置libtorch找不到xxx.dll

项目场景:

使用CLion配置libtorch时遇到该问题


问题描述

使用CLion配置libtorch时,CMakeLists.txt文件写完后,cmake也能成功,但是一旦运行代码就会报错找不到xxx.dll,比如找不到torch_cuda.dll或找不到c10.dll


原因分析:

其实并不是因为你下载的libtorch中没有xxx.dll这些文件,而是因为CLion找不到,你得手动告诉他位置在哪里。这些dll文件都在libtorch/lib下。


解决方案:

在使用CLion运行CMake Application时手动加上环境变量,如下图所示:
在这里插入图片描述
在这里插入图片描述
Environment variables这栏加上libtorch/lib的路径
这时再运行下面代码即可正确加载<torch/torch.h>

下面贴上我的完整的CMakeLists.txt内容:

cmake_minimum_required(VERSION 3.25)
project(torch_cuda_cpp)

set(CMAKE_CXX_STANDARD 17)
set(Torch_DIR E:/CLion/libtorch/share/cmake/Torch)
find_package(Torch REQUIRED)

include_directories(E:/CLion/libtorch/include)
include_directories(E:/CLion/libtorch/include/torch/csrc/api/include)

add_executable(torch_cuda_cpp main.cpp)
target_link_libraries(torch_cuda_cpp ${TORCH_LIBRARIES})
set_property(TARGET torch_cuda_cpp PROPERTY CXX_STANDARD 17)

特别提醒

libtorch分为cuda版和cpu版,其实这和pytorch的cuda版和cpu版是差不多一个一次,只是libtorch是针对C++语言编写的。libtorch的cuda和cpu版的配置方法差不多,网上有很多人介绍怎么配置cpu版的,cuda版的方法也一样,只是别的博客大多只讲了怎么配置,而没有讲遇到的问题怎么解决,就比如我这篇博客里的问题。我踩过的坑记录一下,下面贴一段测试代码:

#include <iostream>
#include <torch/torch.h>

int main() {
    torch::Tensor tensor = torch::rand({2, 3});
    std::cout << tensor << std::endl;
    std::cout << torch::cuda::is_available() << std::endl;
    std::cout << "Hello, World!" << std::endl;
    return 0;
}

在这里插入图片描述

  • 0
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值