欢迎来到代码驿站!

C代码

当前位置:首页 > 软件编程 > C代码

VS2022+libtorch+Cuda11.3安装测试教程详解(调用cuda)

时间:2022-06-18 12:38:47|栏目:C代码|点击:

以下内容默认cuda已经安装完成并添加至系统环境变量

1.下载libtorch

PyTorch

在官网下载压缩包, 可以选择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

libtorch with Cuda 11.3 not linked properly on Windows using Visual Studio 2022 · Issue #72396 · pytorch/pytorch · GitHub

c10::NotImplementedError with minimal example - C++ - PyTorch Forums

上一篇:C语言开发中的常见错误详解

栏    目:C代码

下一篇:C语言的动态内存管理的深入了解

本文标题:VS2022+libtorch+Cuda11.3安装测试教程详解(调用cuda)

本文地址:http://www.codeinn.net/misctech/205177.html

推荐教程

广告投放 | 联系我们 | 版权申明

重要申明:本站所有的文章、图片、评论等,均由网友发表或上传并维护或收集自网络,属个人行为,与本站立场无关。

如果侵犯了您的权利,请与我们联系,我们将在24小时内进行处理、任何非本站因素导致的法律后果,本站均不负任何责任。

联系QQ:914707363 | 邮箱:codeinn#126.com(#换成@)

Copyright © 2020 代码驿站 版权所有