Official PyTorch implementation of Superpoint Transformer introduced in [ICCV'23] "Efficient 3D Semantic Segmentation with Superpoint Transformer" and SuperCluster introduced in [3DV'24 Oral] "Scalable 3D Panoptic Segmentation As Superpoint Graph Clustering"
MIT License
560
stars
72
forks
source link
Fix CUDA version in install.sh to be coherent with pyg wheels (cu 11.8) #40
This PR sets the index url when pip installin torch. This was necessary when installing the env in Google Collab, since torch would be installed with cuda 12.0 while pytorch geometric would be later installed wit cuda 11.8, resulting in an error message.
Before submitting
[X] Did you make sure title is self-explanatory and the description concisely explains the PR?
[X] Did you make sure your PR does only one thing, instead of bundling different changes together?
[X] Did you list all the breaking changes introduced by this pull request?
[ ] Did you test your PR locally with pytest command?
[ ] Did you run pre-commit hooks with pre-commit run -a command?
Still need to test locally, hence the draft PR.
What does this PR do?
This PR sets the index url when pip installin torch. This was necessary when installing the env in Google Collab, since torch would be installed with cuda 12.0 while pytorch geometric would be later installed wit cuda 11.8, resulting in an error message.
Before submitting
pytest
command?pre-commit run -a
command?Did you have fun?
Make sure you had fun coding 🙃