clean-pvnet
clean-pvnet copied to clipboard
Update RANSAC Voting for PyTorch (>1.10) & CUDA (>11.1) Compatibility
โน๏ธ Ensure Compatibility with Newer PyTorch and CUDA Versions for RANSAC Voting Layer
๐ Summary: This commit updates the RANSAC voting module to ensure compatibility with the latest PyTorch and CUDA versions.
๐ Changes Made:
- Pytorch has removed
Thc/thc.h
headers from version >11.1 which gives compilation errors. Commentedextern THCState* state;
inransac_voting.cpp
- Replaced the
torch.solve()
withtorch.linalg.solve()
for newer version. - Applied
tensor.ge()
function on the mask astensor.masked_select
expects a boolean tensor array in newer versions.
๐ Additional Notes:
- Requirements.txt should be updated to work with newer versions as well.
๐ง Dependencies:
- PyTorch: >1.11
- CUDA: >11.1
๐งช Testing:
- Ran comprehensive tests to validate the compatibility with PyTorch [2.1.2] and CUDA [11.8].
- Verified that the RANSAC voting module performs as expected.
๐ Acknowledgments: Thanks to the authors for sharing code for this valuable research.
๐ Date: 21st Jan, 2024