Error importing CPU wheel on machine with CUDA - Githubissues

rusty1s / pytorch_spline_conv

Implementation of the Spline-Based Convolution Operator of SplineCNN in PyTorch

https://arxiv.org/abs/1711.08920

MIT License

172 stars 36 forks source link

Error importing CPU wheel on machine with CUDA #23

Closed aphedges closed 3 years ago

aphedges commented 3 years ago

Looking at __init__.py, it seems that this package as the same issue as https://github.com/rusty1s/pytorch_scatter/issues/229. The fixed used in https://github.com/rusty1s/pytorch_scatter/commit/778f62459ab33497ae13ccbd2d3bb63d7f038020 probably would work here as well.

rusty1s commented 3 years ago

Fixed in https://github.com/rusty1s/pytorch_spline_conv/commit/10bfaf3609c5003ca2b3ecd12e96ec2d54ccd637