VS2022+libtorch+Cuda11.3安装测试教程详解(调用cuda)
以下内容默认cuda已经安装完成并添加至系统环境变量
1.下载libtorch
在官网下载压缩包, 可以选择Release版或者Debug版（根据自己需要）：
下载完成之后选择安装软件的位置进行解压
2.配置VC++目录：
VS新建空项目
2.1添加包含目录：
D:\soft\libtorch\libtorch\include
D:\soft\libtorch\libtorch\include\torch\csrc\api\include
2.2添加库目录：
D:\soft\libtorch\libtorch\lib
3.配置环境变量：
PATH=D:\soft\libtorch\libtorch\lib;%PATH%
4.配置链接器： 4.1链接器--input
D:\soft\libtorch\libtorch\lib\*.lib
4.2链接器--Command Line
/INCLUDE:?warp_size@cuda@at@@YAHXZ /INCLUDE:?_torch_cuda_cu_linker_symbol_op_cuda@native@at@@YA?AVTensor@2@AEBV32@@Z
5.测试配置结果：
#include<torch/torch.h>
#include<torch/script.h>
#include<iostream>
int main() {
std::cout << "cuda::is_available():" << torch::cuda::is_available() << std::endl;
std::cout << "torch::cuda::cudnn_is_available():" << torch::cuda::cudnn_is_available() << std::endl;
std::cout << "torch::cuda::device_count():" << torch::cuda::device_count() << std::endl;
torch::Device device(torch::kCUDA);
torch::Tensor tensor1 = torch::eye(3); // (A) tensor-cpu
torch::Tensor tensor2 = torch::eye(3, device); // (B) tensor-cuda
std::cout << tensor1 << std::endl;
std::cout << tensor2 << std::endl;
}正常结果输出如下：
参考：
Libtorch + vs 2019安装及配置_开拓者5号的博客-CSDN博客_vs2019配置libtorch
Win10+libtorch+CUDA+vs2017_大智若鱼.AI的博客-CSDN博客
win10系统上LibTorch的安装和使用（cuda10.1版本）_*匿名*的博客-CSDN博客_cuda libtorch
c10::NotImplementedError with minimal example - C++ - PyTorch Forums
到此这篇关于VS2022+libtorch+Cuda11.3安装测试（调用cuda）的文章就介绍到这了,更多相关VS2022 libtorch Cuda11.3安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要为大家介绍了C语言实现多项式的相加，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-10-10
字符串函数（String processing function）也叫字符串处理函数，指的是编程语言中用来进行字符串处理的函数，如C，pascal，Visual以及LotusScript中进行字符串拷贝，计算长度，字符查找等的函数2022-04-04
int 转string的方法，用到了特分享一下方便需要的朋友2013-01-01
这篇文章主要为大家详细介绍了在Qt中如何实现读取和写入配置文件的数据，文中的示例代码讲解详细，具有一定的学习价值，感兴趣的小伙伴可以了解一下2023-03-03
这篇文章主要介绍了C++11 lambda表达式/包装器/线程库的相关知识，本文通过示例代码给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2022-05-05
C++ 通过pqxxlib库链接 PostgreSql数据库的详细过程
这篇文章主要介绍了C++ 通过pqxxlib库链接 PostgreSql数据库,本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-04-04
这篇文章主要介绍了c++ 继承的相关资料，文中讲解非常细致，代码帮助大家更好的理解和学习，感兴趣的朋友可以了解下2020-07-07
C语言中的char是用于声明单个字符的关键字,这篇文章主要给大家介绍了关于C语言char类型中存储的相关资料,文中通过实例代码介绍的非常详细,需要的朋友可以参考下2023-01-01
这篇文章主要介绍了15种 C++ 常见报错,本文通过实例代码给大家讲解的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2023-01-01
这篇文章主要介绍了C语言全部内存操作函数的实现详细讲解，作者用图文代码实例讲解的很清晰，有感兴趣的同学可以研究下2021-10-10
最新评论