CUDA 11 support
tengyu-liu opened this issue · 1 comments
tengyu-liu commented
Hi,
I'm encountering some strange errors when using mesh_intersection on 3090 GPU, CUDA 11.1, python 3.8.8, pytorch 1.8.1.
I encountered a CUDA out of memory
error when running examples/batch_smpl_untangle.py
using suggested parameters.
If I add a line val += np.random.normal(0, 0.1, val.shape)
between examples/batch_smpl_untangle.py:129 and 130
, coll_idx
can sometimes have a negative length in loss.py:224
.
It looks like that the problem is with CUDA11. Is it possible for you to support CUDA 11?
Thank you very much!
wangsen1312 commented
I have tried on CUDA 11.3 + python3.7 + pytorch 1.11~ it works well