window安装torch7并运用TH

记录一下安装TH库及自己踩过的坑

安装环境

window10+vs2017
其它:cmake(我也不记得什么版本了,反正比较新)
我一开始用ubuntu编译了一个.so也是晕乎乎啊,注意,在linux运行的话,在ubuntu下进行编译,如果要在windows跑,就在window下进行编译。

安装torch

这里有一个官方介绍pytorch快速浏览
1.先找到源代码,clone下来,我是clone到window下。
2.打开cmake
在这里插入图片描述
其中第3步选择自己电脑上的vs版本及相应的位数。
在这里插入图片描述
如果第5步不能正确打开相对应的工程文件,可以在build目录下找到Project.sln选择合适的vs版本编译器打开。
这里建议选择release进行编译。debug我就报了下面这个错误。

error D8016: “/O2”和“/RTC1”命令行选项不兼容的解决方案

编译完成后
将build目录下general.h拷贝至torch7/lib/TH
在这里插入图片描述

在这里插入图片描述
以上的TH就是我们需要的inlude目录

另外release目录下是我们需要的lib文件
在这里插入图片描述

尝试运行

在这里插入图片描述
在这里插入图片描述

首先设置工程中对应的include位置,lib库位置和lib文件
官方的测试代码

#include "TH/TH.h"
#pragma comment(lib,"TH.lib")
int main()
{
    THFile *x_file = THDiskFile_new("x", "r", 0);
    THFile *y_file = THDiskFile_new("y", "r", 0);

    THFloatTensor *x = THFloatTensor_newWithSize1d(10);
    THFloatTensor *y = THFloatTensor_newWithSize1d(10);

    THFile_readFloat(x_file, x->storage);
    THFile_readFloat(y_file, y->storage);

    double result = THFloatTensor_dot(x, y) + THFloatTensor_sumall(x);

    printf("%f\n", result);

    THFloatTensor_free(x);
    THFloatTensor_free(y);
    THFile_free(x_file);
    THFile_free(y_file);
    return 0;
}
  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值