FateScript / CenterNet-better

An easy to understand and better performance version of CenterNet
Apache License 2.0
548 stars 104 forks source link

error in modulated_deformable_col2im_coord_cuda: invalid device function #32

Closed kritohyh closed 3 years ago

kritohyh commented 3 years ago

Hello, could you help me? I am running the code with the following problem:

error in modulated_deformable_col2im_coord_cuda: invalid device function
error in modulated_deformable_col2im_cuda: invalid device function
error in modulated_deformable_im2col_cuda: invalid device function

Environmental information that the program prints:


sys.platform linux Python 3.7.9 (default, Aug 31 2020, 12:42:55) [GCC 7.3.0] Numpy 1.19.4 dl_lib Compiler GCC 5.4 dl_lib CUDA Compiler 10.0 dl_lib_ENV_MODULE PyTorch 1.4.0+cu100 PyTorch Debug Build False torchvision 0.5.0+cu100 CUDA available True GPU 0,1,2,3,4,5,6,7 GeForce RTX 2080 Ti CUDA_HOME /usr/local/cuda NVCC Cuda compilation tools, release 10.0, V10.0.130 Pillow 8.0.1 cv2 4.4.0


PyTorch built with:

bkj commented 3 years ago

+1 -- I'm getting the same error -- any ideas on how to fix?

FateScript commented 3 years ago

This error occurred using cuda 10.0. Please upgrade your cuda, thanks.