Visual Studio配置libtorch(cuda安装一步到位)

visual Studio安装

visual Studio点击安装

具体的安装和配置过程这里就不进行细讲了,可以参考我这篇博客Visual Studio配置OpenCV(保姆及教程)

在这里插入图片描述

cuDNN安装

cuDNN点击安装

查看自己的显卡驱动版本以确定cuDNN的安装版本,一般是最新版本的就没什么问题。

在这里插入图片描述

直接选择最新的进行安装,具体安装过程这里就不进行说明了。

在这里插入图片描述

CUDAToolkit安装

CUDAToolkit点击安装

选择配套的CUDAToolkit,这里需要CUDA的12.x版本,在下方红框中随便选择一个安装即可

在这里插入图片描述

这里将CUDAToolkit进行解压,会出现以下目录,将其全部复制粘贴到cuDNN所在的目录下,一般默认安装位置为C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.3

在这里插入图片描述

进入环境变量将以下路径添加至环境变量中

在这里插入图片描述

打开窗口【win +R】输入cmd,再输入nvidia-smi显示以下界面就说明CUDA安装成功了。

在这里插入图片描述

libtorch下载

libtorch点击下载

我们直接选择cuda版本的debug版本进行下载。需要注意的是debug和rellease版本是有区别的。

在这里插入图片描述

下载完成之后解压将以下路径放置环境变量中

在这里插入图片描述

Visual Studio配置libtorch(cuda版本配置)

打开属性进行配置

在这里插入图片描述

点击调试,点击环境,添加路径至环境
在这里插入图片描述
点击VC++目录, 点击包含目录, 将libtorch和cuda的路径进行添加。
在这里插入图片描述
点击VC++目录, 点击库目录,将libtorch和cuda路径进行添加。
在这里插入图片描述
点击c/c++,点击常规,点击附加包含目录,添加$(IncludePath),将SDL检查改为否
在这里插入图片描述
点击语言,点击c++语言标准,改为c++17, 根据自己报错进行更改,我的是c++17,默认为c++14.将符合模式改为否
在这里插入图片描述
点击链接器,点击输入,点击附加依赖项,将你安装的libtorch路径E:\DownloadTemp\libtorch\lib下的所有.lib文件进行添加。
在这里插入图片描述
到这里你的libtorch的cpu版本就已经配置完成了,如果需要配置gpu版本就跟以下这个链接器的命令行有关系,在命令行的其他选型中输入/INCLUDE:"?ignore_this_library_placeholder@@YAHXZ" ,不同的CUDA版本需要不同的命令行,根据自己的版本进行配置。
在这里插入图片描述

输入以下代码进行测试

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

using namespace std;

int main() {
	auto a = torch::rand({ 5, 3 });
	if (torch::cuda::is_available()) {
		cout << "cuda可用" << endl;
	}
	else
	{
		cout << "不可用" << endl;
	}
	cout << a << endl;
}

显示以下界面说明你的环境以配置成功

在这里插入图片描述

最近更新

  1. TCP协议是安全的吗?

    2024-03-17 06:42:01       18 阅读
  2. 阿里云服务器执行yum,一直下载docker-ce-stable失败

    2024-03-17 06:42:01       19 阅读
  3. 【Python教程】压缩PDF文件大小

    2024-03-17 06:42:01       18 阅读
  4. 通过文章id递归查询所有评论(xml)

    2024-03-17 06:42:01       20 阅读

热门阅读

  1. 计算最长的字符串长度

    2024-03-17 06:42:01       20 阅读
  2. HTTP和HTTPS的区别,HTTPS加密原理是?

    2024-03-17 06:42:01       20 阅读
  3. 渗透测试与HTTP中的PUT请求

    2024-03-17 06:42:01       18 阅读
  4. 深度学习pytorch——索引与切片

    2024-03-17 06:42:01       21 阅读
  5. PyTorch究竟是什么?

    2024-03-17 06:42:01       23 阅读
  6. PyTorch学习笔记之基础函数篇(十一)

    2024-03-17 06:42:01       18 阅读
  7. 微信小程序上传图片c# asp.net mvc端接收案例

    2024-03-17 06:42:01       21 阅读
  8. Spring核心方法:Refresh全解(WebMVC如何装配、关联)

    2024-03-17 06:42:01       21 阅读
  9. 阿里提前批(阿里云)一面30min

    2024-03-17 06:42:01       24 阅读
  10. MATLAB中的矩阵和数组,它们之间有什么区别?

    2024-03-17 06:42:01       21 阅读
  11. C语言每日一题—判断是否为魔方矩阵

    2024-03-17 06:42:01       20 阅读
  12. C++ 基础组件(1)定时器

    2024-03-17 06:42:01       18 阅读