安装环境
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;
}