No module named simple_knn

Hi, I've successfully compiled and installed simple_knn under the directory submodules/simple_knn using pip install -e .. But when I run python train.py the gaussian_model.py in scene complains that it can not find simple_knn package, however, I pip show simple_knn, it does exist. What could be the problem?