unlimblue / KNN_CUDA

pytorch knn [cuda version]
292 stars 37 forks source link

import fails on google colab #4

Closed j-bac closed 4 years ago

j-bac commented 4 years ago

On google colab import fails:

I installed with:

!git clone https://github.com/unlimblue/KNN_CUDA.git
!cd KNN_CUDA && make && make install
!wget https://github.com/ninja-build/ninja/releases/download/v1.8.2/ninja-linux.zip
!sudo unzip ninja-linux.zip -d /usr/local/bin/
!sudo update-alternatives --install /usr/bin/ninja ninja /usr/local/bin/ninja 1 --force 
ImportError: No module named 'knn'
unlimblue commented 4 years ago

Try pip and make sure ninja installed successfully.