问题
error: identifier “THCState_getCurrentStream” is undefined
…
AT_ASSERTM(trans.type().is_cuda(), “trans must be a CUDA tensor”);
在对DCNv2模型进行编译的时候出现了报错,记录一下解决办法。
解决
1、Google:
似乎是 pytorch 版本的问题!好不容易才安装上啊!!!
2、yolact模型DCNv2模块编译错误解决方法
参考:
error: identifier “THCState_getCurrentStream” is undefined
…
AT_ASSERTM(trans.type().is_cuda(), “trans must be a CUDA tensor”);
在对DCNv2模型进行编译的时候出现了报错,记录一下解决办法。
1、Google:
似乎是 pytorch 版本的问题!好不容易才安装上啊!!!
2、yolact模型DCNv2模块编译错误解决方法
参考:
您可能感兴趣的与本文相关的镜像
PyTorch 2.5
PyTorch 是一个开源的 Python 机器学习库,基于 Torch 库,底层由 C++ 实现,应用于人工智能领域,如计算机视觉和自然语言处理

被折叠的 条评论
为什么被折叠?