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

2024-03-17 03:12

本文主要是介绍Visual Studio配置libtorch(cuda安装一步到位),希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

Visual Studio配置libtorch

  • visual Studio安装
  • cuDNN安装
  • CUDAToolkit安装
  • libtorch下载
  • 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;
}

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

在这里插入图片描述

这篇关于Visual Studio配置libtorch(cuda安装一步到位)的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/817629

相关文章

Zookeeper安装和配置说明

一、Zookeeper的搭建方式 Zookeeper安装方式有三种,单机模式和集群模式以及伪集群模式。 ■ 单机模式:Zookeeper只运行在一台服务器上,适合测试环境; ■ 伪集群模式:就是在一台物理机上运行多个Zookeeper 实例; ■ 集群模式:Zookeeper运行于一个集群上,适合生产环境,这个计算机集群被称为一个“集合体”(ensemble) Zookeeper通过复制来实现

CentOS7安装配置mysql5.7 tar免安装版

一、CentOS7.4系统自带mariadb # 查看系统自带的Mariadb[root@localhost~]# rpm -qa|grep mariadbmariadb-libs-5.5.44-2.el7.centos.x86_64# 卸载系统自带的Mariadb[root@localhost ~]# rpm -e --nodeps mariadb-libs-5.5.44-2.el7

Centos7安装Mongodb4

1、下载源码包 curl -O https://fastdl.mongodb.org/linux/mongodb-linux-x86_64-rhel70-4.2.1.tgz 2、解压 放到 /usr/local/ 目录下 tar -zxvf mongodb-linux-x86_64-rhel70-4.2.1.tgzmv mongodb-linux-x86_64-rhel70-4.2.1/

hadoop开启回收站配置

开启回收站功能,可以将删除的文件在不超时的情况下,恢复原数据,起到防止误删除、备份等作用。 开启回收站功能参数说明 (1)默认值fs.trash.interval = 0,0表示禁用回收站;其他值表示设置文件的存活时间。 (2)默认值fs.trash.checkpoint.interval = 0,检查回收站的间隔时间。如果该值为0,则该值设置和fs.trash.interval的参数值相等。

NameNode内存生产配置

Hadoop2.x 系列,配置 NameNode 内存 NameNode 内存默认 2000m ,如果服务器内存 4G , NameNode 内存可以配置 3g 。在 hadoop-env.sh 文件中配置如下。 HADOOP_NAMENODE_OPTS=-Xmx3072m Hadoop3.x 系列,配置 Nam

wolfSSL参数设置或配置项解释

1. wolfCrypt Only 解释:wolfCrypt是一个开源的、轻量级的、可移植的加密库,支持多种加密算法和协议。选择“wolfCrypt Only”意味着系统或应用将仅使用wolfCrypt库进行加密操作,而不依赖其他加密库。 2. DTLS Support 解释:DTLS(Datagram Transport Layer Security)是一种基于UDP的安全协议,提供类似于

Centos7安装JDK1.8保姆版

工欲善其事,必先利其器。这句话同样适用于学习Java编程。在开始Java的学习旅程之前,我们必须首先配置好适合的开发环境。 通过事先准备好这些工具和配置,我们可以避免在学习过程中遇到因环境问题导致的代码异常或错误。一个稳定、高效的开发环境能够让我们更加专注于代码的学习和编写,提升学习效率,减少不必要的困扰和挫折感。因此,在学习Java之初,投入一些时间和精力来配置好开发环境是非常值得的。这将为我

【Python编程】Linux创建虚拟环境并配置与notebook相连接

1.创建 使用 venv 创建虚拟环境。例如,在当前目录下创建一个名为 myenv 的虚拟环境: python3 -m venv myenv 2.激活 激活虚拟环境使其成为当前终端会话的活动环境。运行: source myenv/bin/activate 3.与notebook连接 在虚拟环境中,使用 pip 安装 Jupyter 和 ipykernel: pip instal

安装nodejs环境

本文介绍了如何通过nvm(NodeVersionManager)安装和管理Node.js及npm的不同版本,包括下载安装脚本、检查版本并安装特定版本的方法。 1、安装nvm curl -o- https://raw.githubusercontent.com/nvm-sh/nvm/v0.39.0/install.sh | bash 2、查看nvm版本 nvm --version 3、安装

如何在Visual Studio中调试.NET源码

今天偶然在看别人代码时,发现在他的代码里使用了Any判断List<T>是否为空。 我一般的做法是先判断是否为null,再判断Count。 看了一下Count的源码如下: 1 [__DynamicallyInvokable]2 public int Count3 {4 [__DynamicallyInvokable]5 get