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

No module named 'roi_align_api' #4

Open cyj95 opened 1 year ago

cyj95 commented 1 year ago

Make sure your device is CUDA enabled. Build and install source code of roi_align_api and rod_align_api.

how to do this step? can you show the file of yours?

cyj95 commented 1 year ago

make_all.sh

Brembles commented 11 months ago

确保您的设备启用了 CUDA。构建并安装 roi_align_api 和 rod_align_api 的源代码。

这一步该怎么做呢？你能展示一下你的文件吗？

How did you do this step to build and install the source code of roi_align_api and rod_align_api? I kept getting errors and the path to the THC/THC.h file was missing.

intchous commented 8 months ago

The "make_all.sh" script depends on two files in "rod_align/make.sh" and "roi_align/make.sh". You need to adjust the -arch parameter for your GPU. For example, a 2080ti GPU, with a Compute Capability of 7.5 (found on https://developer.nvidia.com/cuda-gpus), requires setting -arch=sm_75. For a 4090 GPU, it should be -arch=sm_89. Also, update the CUDA_HOME path to match your CUDA installation directory.