No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda'

lucasjinreal / DCNv2_latest

DCNv2 supports decent pytorch such as torch 1.5+ (now 1.8+)

BSD 3-Clause "New" or "Revised" License

637 stars 124 forks source link

No CUDA runtime is found, using CUDA_HOME='/usr/local/cuda' #61

Open mic2112 opened 2 years ago

mic2112 commented 2 years ago

Hey @jinfagang,

I'm trying to run and compile DCNv2, on the latest version of pytorch .i.e. 1.12 with a cuda version of 11.1. I get the following warning when running build:

and testgpu.py returns with the following output:

Can you help, please? TIA