VS2022+libtorch+Cuda11.3安裝測試教程詳解(調(diào)用cuda)
以下內(nèi)容默認(rèn)cuda已經(jīng)安裝完成并添加至系統(tǒng)環(huán)境變量
1.下載libtorch
在官網(wǎng)下載壓縮包, 可以選擇Release版或者Debug版（根據(jù)自己需要）：
下載完成之后選擇安裝軟件的位置進(jìn)行解壓
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.配置環(huán)境變量：
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.測試配置結(jié)果：
#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;
}正常結(jié)果輸出如下：
參考：
Libtorch + vs 2019安裝及配置_開拓者5號的博客-CSDN博客_vs2019配置libtorch
Win10+libtorch+CUDA+vs2017_大智若魚.AI的博客-CSDN博客
win10系統(tǒng)上LibTorch的安裝和使用（cuda10.1版本）_*匿名*的博客-CSDN博客_cuda libtorch
c10::NotImplementedError with minimal example - C++ - PyTorch Forums
到此這篇關(guān)于VS2022+libtorch+Cuda11.3安裝測試（調(diào)用cuda）的文章就介紹到這了,更多相關(guān)VS2022 libtorch Cuda11.3安裝內(nèi)容請搜索腳本之家以前的文章或繼續(xù)瀏覽下面的相關(guān)文章希望大家以后多多支持腳本之家！
相關(guān)文章
格式化輸出在C語言中非常常用,提供了多種用法來控制輸出的格式,下面這篇文章主要給大家介紹了關(guān)于C語言中格式化輸出符號%d、%c、%p、%x等的相關(guān)資料,文中通過代碼介紹的非常詳細(xì),需要的朋友可以參考下2024-06-06
在日常編程中，對于字符串的操作中我們都很習(xí)慣的使用一些C庫中自帶的函數(shù)，并且關(guān)于字符串的操作函數(shù)基本都在 string.h 這個頭文件中。關(guān)于C庫字符串常用處理函數(shù)也是平時面試或者考試過程中非常喜歡考的，本文將帶大家手動來實(shí)現(xiàn)這些常用函數(shù)2022-05-05
這篇文章主要介紹了C語言字符串快速壓縮算法代碼，將字符串中連續(xù)出席的重復(fù)字母進(jìn)行壓縮，其主要的壓縮字段的格式為”字符重復(fù)的次數(shù)+字符”。有需要的小伙伴參考下吧。2015-03-03
c++實(shí)現(xiàn)reactor高并發(fā)服務(wù)器的詳細(xì)教程
這篇文章主要介紹了c++從零實(shí)現(xiàn)reactor高并發(fā)服務(wù)器,包括環(huán)境準(zhǔn)備和基礎(chǔ)知識介紹,本文給大家介紹的非常詳細(xì),感興趣的朋友跟隨小編一起看看吧2024-03-03
本篇文章是對C語言實(shí)現(xiàn)shell指令的方法進(jìn)行了詳細(xì)的分析介紹，需要的朋友參考下2013-05-05
Matlab實(shí)現(xiàn)別踩白塊小游戲的示例代碼
別踩白塊是一款音樂類休閑游戲，游戲的玩法不難，只需跟著音樂的節(jié)奏點(diǎn)中對的方塊即可。本文將用Matlab實(shí)現(xiàn)這一經(jīng)典游戲，感興趣的可以了解一下2022-03-03
大家可能對這個臨時對象這個概念還不是很清楚，那么首先我們花一些時間來理解臨時對象2013-01-01
C++是一門面向?qū)ο蟮木幊陶Z言，其中的派生機(jī)制是其重要的面向?qū)ο筇匦灾?。本文我們就來詳?xì)地學(xué)習(xí)一下C++中的派生機(jī)制的相關(guān)知識吧2023-04-04
這篇文章主要介紹了C++之&與*符號用法案例詳解,本篇文章通過簡要的案例,講解了該項技術(shù)的了解與使用,以下就是詳細(xì)內(nèi)容,需要的朋友可以參考下2021-09-09
這篇文章主要為大家詳細(xì)介紹了C語言模擬實(shí)現(xiàn)簡單掃雷游戲，文中示例代碼介紹的非常詳細(xì)，具有一定的參考價值，感興趣的小伙伴們可以參考一下2019-10-10