# Build Deformable Convolution cd mm_modules/DCN python setup.py build install

ziplab / LITv2

[NeurIPS 2022 Spotlight] This is the official PyTorch implementation of "Fast Vision Transformers with HiLo Attention"

Apache License 2.0

229 stars 11 forks source link

# Build Deformable Convolution cd mm_modules/DCN python setup.py build install #4

Closed renqi1998 closed 2 years ago

renqi1998 commented 2 years ago

Hello, when I set up a compilation environment for LITv2, I followed your steps and ran python setup.py build install. I encountered an error that there is no cuda, but there is CUDA in the environment. How to solve this?

HubHop commented 2 years ago

Hi @renqi1998, it seems like your python environment cannot find the path to CUDA. E.g. CUDA_HOME in your system-level configuration. You can check your CUDA installation by using nvcc -V in your terminal. Please make sure you have successfully installed cuda in your operating system.

renqi1998 commented 2 years ago

Thank you, I used this statement “python3 setup.py install” to compile successfully