_nms.pyd 编译出错问题记录(二)
前面的_nms.pyd 编译通过了,是在 pytorch 0.4.0 的基础上编译的,结果项目后面 出现 不能带 ‘.’ 的错误,还是 跟随作者回退到 pytorch 0.3.0 。
pytorch 0.3.0 中的TH相关函数实现在 TH.dll ,但是没有 TH.lib文件,所以_nms.pyd 编译又不通过,而且查看还发现 是生成 _nms.so文件,于是改 ffi 原始函数图示如下:
#libname = cffi_wrapper_name + '.so'
ext_suf = '.pyd' if os.sys.platform == 'win32' else '.so'
libname = cffi_wrapper_name + ext_suf
这次是生成_nms.pyd文件,但是还是 链接不上 TH的那些函数,于是从官网下载了 pytorch 0.3.0 的源码,vs命令工具 cd 到 \pytorch-0.3.0\torch\lib\TH 目录下,
新建build文件,cmake编译 生成TH的工程文件,在打开工程文件更改release 64 编译生成 TH.dll ,TH.lib. 如下:
·
开始直接复制过去,可以成功生成_nms.pyd,但是TH.dll出现错误,程序在pytorch前向运算(forward)时崩溃。提示错误
Process finished with exit code -1073741819 (0xC0000005)
于是用了以前的 TH.dll ,用自己生成 的TH.lib来编译,再次成功生成_nms.pyd并且程序运行顺利。