Point-M2AE
Point-M2AE copied to clipboard
question about the pytorch version
I'm wondering your pytorch version. My torch vision is 2.0.0. It seems that this version will have problem about the complication of the emd_kernal.cu because of the disappearance of the THC/THC.h. Looking forward to your reply.
He might use a Pytorch version lower than 1.8 because THCuda was deprecated since. If you want to use the newest pytorch version you can:
- Remove the
#include <THC/THC.h>since it's deprecated (https://github.com/pytorch/pytorch/pull/66391) and replace allTHCudaCheckbyC10_CUDA_CHECK. - Change all the
CHECK_EQtoTORCH_CHECK_EQsince the alias name as changed (https://github.com/pytorch/pytorch/pull/82032)
Have fun 🤩
@ZrrSkywalker If you want I can make a PR to fix this, let me know