Dependencies

[Phoveran]

Thanks for your work!

Could you please provide details on your dependencies? I got the following error

name 'to_cupy' is not defined

I'm using pytorch 1.10.0 and cuda 11.3

Thanks!

[sebastienloustau]

yo need to install cupy here https://github.com/cupy/cupy