search

longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch
555 stars 103 forks source link

GPU support #3

Closed simo23 closed 6 years ago

simo23 commented 6 years ago

Hi, thank you very much for this code!

The code works fine for me if I change the capability of the gpu in make.sh as:

$CUDA_PATH/bin/nvcc -c -o crop_and_resize_kernel.cu.o crop_and_resize_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_50

Because I have a GPU with 5.0 capability which is slightly less than sm_52 as you wrote. Maybe it is worth mentioning it in the README to check the capability of the GPU before building.

longcw commented 6 years ago

Already added it :-D

simo23 commented 6 years ago

Thanks!