search

sniklaus / pytorch-pwc

a reimplementation of PWC-Net in PyTorch that matches the official Caffe version
GNU General Public License v3.0
608 stars 122 forks source link

Can it test in CPU device? #46

Closed bugczw closed 3 years ago

bugczw commented 3 years ago

In the project, in the file correlation/correlation.py, it uses a module called cupy. However, cupy is based on cuda, which means that it needs to contain GPU. However, in most macOS computers, cuda and cupy cannot be supported. So, if I want to test this project without cuda, is there any way to achieve？Or is there any method that does not use cupy module to implement the correlation function?

sniklaus commented 3 years ago

To make it run on a CPU, you would have to rewrite correlation.py accordingly. For more details, please see #39