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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了EasyC++内部链接性和无链接性，当我们使用static关键字，将变量的作用于限制在整个文件时，该变量的链接性为内部链接性，然而无链接性的变量其实就是在代码块当中使用static关键字创建的，接下来一起进入文章了解更多内容吧2021-12-12
这篇文章主要给大家介绍了关于C语言中递归的相关资料，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-11-11
这篇文章主要为大家详细介绍了C语言实现哈夫曼树的构建，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-04-04
这篇文章主要为大家详细介绍了C语言实现消消乐小游戏，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2020-12-12
这篇文章主要为大家详细介绍了C语言链表实现学生成绩管理系统，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2022-07-07![深入理解C语言的new[]和delete[]](//img.jbzj.com/images/xgimg/bcimg5.png)
new和delete既是C++中的关键字也是一种特殊的运算符。这篇文章主要介绍了C++的new和delete详解,需要的朋友可以参考下2021-09-09
这篇文章主要为大家详细介绍了C语言实现24点游戏源代码，文中示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下2019-10-10
GDI+（Graphics Device Interface Plus）是一种用于图形绘制和图像处理的应用程序编程接口（API）,在Windows平台上广泛使用,本文就来介绍一下如何使用GDI实现图片格式转换吧2023-12-12
这篇文章主要介绍了线程按指定顺序输出字符到数组的实例代码,需要的朋友可以参考下2014-02-02
这篇文章主要介绍了c语言5个常用的排序算法实例代码，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-02-02
![深入理解C语言的new[]和delete[]](http://img.jbzj.com/images/xgimg/bcimg5.png)
最新评论