VScode + libtorch配置

本文详细指导如何在Linux环境下配置VScode以使用libtorch,包括下载libtorch,安装必要的依赖,配置CMakeLists.txt、launch.json、tasks.json和c_cpp_properties.json文件以实现C++代码的编译和调试。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

配置选项:基于Linux,VScode + libtorch

步骤一:libtorch下载(DEBUG版本)
前往libtorch官网https://pytorch.org/get-started/locally/,选择对应版本的libtorch进行下载
解压完成后,放入D盘或者其他位置,记录其路径即可。

步骤二:安装VScode(这边不再赘述),安装完成后,从官网下载cmake进行安装,这边不推荐通过VScode自带的Extensions增加cmake和cmake tools模块。

步骤三:安装好VScode后,新建项目,然后新建main.cpp,然后逐个配置CMakeLists.txt,launch.json,tasks.json,c_cpp_properties.json文件。

main.cpp 代码如下:

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

using namespace std;
int main()
{
    torch::Tensor output;
    cout << "cuda is_available: " << torch::cuda::is_available() << endl
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值