Prediction use pretrained network on CPU

haofeixu / aanet

[CVPR'20] AANet: Adaptive Aggregation Network for Efficient Stereo Matching

Apache License 2.0

524 stars 102 forks source link

Closed LongshengDu closed 4 years ago

LongshengDu commented 4 years ago

Can I build deformable convolution without cuda and run pretrained network on CPU? Just want to do some tests and currently not have a nvidia card.

When I run predict.py, I got

ImportError: cannot import name 'deform_conv_cuda' from 'nets.deform_conv'

When I build deformable convolution, I got

AssertionError: Torch not compiled with CUDA enabled

haofeixu commented 4 years ago