Closed Pinnh closed 4 years ago
how to solve this issue with pytorch 0.4.0 and python3?
so simple ,upgrade your pytorch version
to which one?
pytorch1.2.0+cuda10.0 works
I'm using pytorch 1.9.1+cuda11.1 and changing AT_CHECK to TORCH_CHECK actually worked. Thank you for your help.
In higher version torch , that AT_CHECK function is changed to TORCH_CHECK