search

intchous / SmartText

Code for the paper "Harmonious Textual Layout Generation over Natural Images via Deep Aesthetics Learning" (TMM 2021)
MIT License
37 stars 3 forks source link

src/rod_align_cuda.cpp:1:21: fatal error: THC/THC.h: No such file or directory #3

Open Du-Yao opened 1 year ago

Du-Yao commented 1 year ago

我使用的pytorch版本是1.13，cuda11.6，在编译rod_align 和roi_align时报错，如题目，请问该如何解决。

intchous commented 1 year ago

也许是因为新版的 pytorch 移除了 THC files, 尝试把 pytorch 版本降到 1.10 以下看看

cyj95 commented 1 year ago

pytorch1.6也不行啊

cyj95 commented 12 months ago

fatal error: THC/THC.h: No such file or directory 降低版本没有效果，有什么其他的方法吗

undcloud commented 6 months ago

我用docker环境可以跑通，好像仍旧有这个编译问题，但是程序能运行，就很奇怪，经验分享给大家，，，

docker pull nvidia/cuda:11.1.1-cudnn8-devel-ubuntu20.04 pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html