机器学习 pytorch c++ windows 环境搭建

1.下载libtorch
(1)Pytorch 下载官网:https://pytorch.org/get-started/locally/
(2)选择 版本为最新的1.3版,操作系统为windows,包为libtorch,语言为c++,cuda 版本为最新的10.1。如下图:
机器学习 pytorch c++ windows 环境搭建_第1张图片
(3)点击release 版的libtorch 库的下载链接,下载压缩包。
2.解压下载到的压缩文件到任意目录
3.下载安装CUDA 10.1
(1)需要英伟达的支持cuda的显卡,我的显卡是GTX950M。如下图:
机器学习 pytorch c++ windows 环境搭建_第2张图片
4.测试代码如下:
#include
#include
using namespace std;

int main()
{
torch::Tensor tensor = torch::rand({ 2, 3 });

std::cout << tensor.cuda() << std::endl;

}
5.Vs2019 工程环境配置:
(1)选择release x64
(2)Include 目录:
机器学习 pytorch c++ windows 环境搭建_第3张图片
(3)Lib 目录:
机器学习 pytorch c++ windows 环境搭建_第4张图片
(4)Lib库:
在这里插入图片描述
6.结果
在这里插入图片描述

你可能感兴趣的:(深度学习,机器学习)