diff --git a/install.sh b/install.sh index 2a3e2dbbcfcc60596d8a878616f76e06da44e827..2cb951ac9f79b6c3d934d189bab2ee533fbb629d 100644 --- a/install.sh +++ b/install.sh @@ -16,6 +16,8 @@ pip install sharedarray tensorboard tensorboardx yapf addict einops scipy plyfil pip install torch-scatter torch-sparse torch-cluster pip install torch-geometric +pip install spconv-cu120 + # PTv1 & PTv2 or precise eval cd libs/pointops