What's your version of pytorch, when you implement the ROI Align with numba?

[ywcmaike]

What's your version of pytorch and numba, when you implement the ROI Align with numba?
Hello, I want to accelerate my code with numba using pytorch, and I found the example of ROI Align suit to me, and I want to learn your code, but encounter some issue. Just as followed.
[1] setattr(tensor, 'cuda_array_interface', cai_dict)
AttributeError: can't set attribute
or
[2] AttributeError: module 'numba' has no attribute 'as_cuda_array'
or AttributeError: module 'numba.cuda' has no attribute 'as_cuda_array'

my environment is as followed：

torch.version
'1.2.0'
import numba
numba.version
'0.46.0'

I hope to receive your reply.
What's your version of pytorch and numba, when you implement the ROI Align with numba?
@dandelin