【vs2022安装libtorch——gpu版本安装过程】

vs2022安装libtorch——gpu版本,历尽辛苦终于搞定了(呜呜呜)

—————————————————————————记录安装过程—————————————————————————

1、安装好cuda(这一步自己的gpu不同,找资料自己安装)

2、在电脑桌面,win+r输入cmd,输入nvcc --version查看自己的cuda版本号——如下:

版本号
这很重要)注:这里是release版本,则libtorch要下载release版本的gpu压缩包,其次就是版本号,这里选择2.3.0和cu11.8,但不支持cuda11.0的所有没用,就找了其他版本libtorch-win-shared-with-deps-1.10.0+cu102.zip(1.10.0、cu10.2)就支持了,cu版本是高版本向下兼容的
包

3、打开vs2022,新建空项目,建好项目后右键点击项目名称,弹出对话框,点击属性,进入项目属性页把:配置:改成release,平台:x64

项目属性

(1)点击vc++目录

在包含目录中添加,解压出来的:
your_path\libtorch\include和your_path\libtorch\include\torch\csrc\api\include
在库目录中添加,解压出来的:
your_path\libtorch\lib

目录

(2)在配置属性里面点击调试

在环境中添加,解压出来的:your_path\libtorch\lib;
这是我的:PATH=G:\libtorch\libtorch1.10.0+cu102\libtorch\lib;%PATH%
模板:PATH=your_path\libtorch\lib;%PATH%

调试

(3)点击连接器

点击——常规,在附加库目录中添加,解压出来的:your_path\libtorch\lib
这是我的:G:\libtorch\libtorch1.10.0+cu102\libtorch\lib
点击——输入,在附加依赖项中添加,解压出来的:your_path\libtorch\lib*.lib
这是我的:G:\libtorch\libtorch1.10.0+cu102\libtorch\lib*.lib

点击——命令行,在其他选项下边输入:命令字符串
注:每个libtorch版本有不同的命令,这里自己找到自己版本的命令复制粘贴即可
这是我的:
/INCLUDE:“?ignore_this_library_placeholder@@YAHXZ”

连接器
模板22
命令行

4、以上的每一步配置好了都要点击页面右下角的确定,

进行代码测试:
代码

运行代码成功会出现如下结果表明安装成功:

成功

测试代码:

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

int main() {
	torch::Tensor tensor = torch::rand({ 2, 3 });
	if (torch::cuda::is_available()) {
		std::cout << "CUDA is available! Training on GPU" << std::endl;
		auto tensor_cuda = tensor.cuda();
		std::cout << tensor_cuda << std::endl;
	}
	else
	{
		std::cout << "CUDA is not available! Training on CPU" << std::endl;
		std::cout << tensor << std::endl;
	}

	std::cin.get();
}

安装过程踩了无数坑,最主要的是版本要找好,其他的都比较简单!!! 记得点赞收藏转发给你需要的朋友!

  • 20
    点赞
  • 16
    收藏
    觉得还不错? 一键收藏
  • 2
    评论
以下是安装tensorflow-gpu的步骤: 1. 安装CUDA Toolkit和cuDNN库 TensorFlow GPU版本需要CUDA Toolkit和cuDNN库的支持。请根据您的系统环境下载适合的版本。推荐使用CUDA 10.0和cuDNN 7.4.2。下载链接如下: - CUDA Toolkit 10.0:https://developer.nvidia.com/cuda-10.0-download-archive - cuDNN 7.4.2:https://developer.nvidia.com/cudnn 安装CUDA Toolkit和cuDNN库时,请注意设置环境变量,以便TensorFlow能够找到它们。 2. 创建conda虚拟环境 您可以使用conda创建一个虚拟环境来安装和管理TensorFlow GPU版本以及其他软件包。打开终端并运行以下命令: ```bash conda create -n tensorflow-gpu python=3.7 ``` 这个命令将创建一个名为“tensorflow-gpu”的conda虚拟环境,其中Python的版本是3.7。 3. 激活conda虚拟环境 使用以下命令激活conda虚拟环境: ```bash conda activate tensorflow-gpu ``` 4. 安装TensorFlow GPU版本 通过以下命令安装TensorFlow GPU版本: ```bash pip install tensorflow-gpu==2.0.0 ``` 这个命令将安装TensorFlow 2.0.0 GPU版本。 5. 验证安装 要验证TensorFlow GPU是否正确安装,请使用以下Python代码: ```python import tensorflow as tf print(tf.test.is_gpu_available()) ``` 如果输出的结果为True,则表示TensorFlow GPU版本安装正确。 提示:安装TensorFlow GPU版本之前,请确保您的GPU支持CUDA。您可以在以下链接中查找支持CUDAGPU:https://developer.nvidia.com/cuda-gpus
评论 2
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值