search

longcw / RoIAlign.pytorch

RoIAlign & crop_and_resize for PyTorch
555 stars 103 forks source link

support pytorch-0.4.1 #13

Closed lxtGH closed 6 years ago

lxtGH commented 6 years ago

It seems like some API changes in pytorch-0.4.1

crazysal commented 5 years ago

This does not run on CPU now. Earlier version was running on CPU. Updated pytorch to -> 0.4.1 : applied this patch. now the following error : 

/lib/python3.5/site-packages/torch/utils/ffi/../../lib/include/THC/THCGeneral.h:12:10: fatal error: cuda.h: No such file or directory
 #include "cuda.h"
          ^~~~~~~~

How to run on cpu with pytorch v=0.4.1