diff --git a/environment.yml b/environment.yml index fdae5d1..c68b52f 100755 --- a/environment.yml +++ b/environment.yml @@ -4,7 +4,7 @@ channels: - conda-forge - defaults dependencies: - - cudatoolkit=12.1 + - cudatoolkit - plyfile=0.8.1 - python=3.10 - pip @@ -24,4 +24,4 @@ dependencies: - scipy - plyfile - recordclass - - scikit-image \ No newline at end of file + - scikit-image diff --git a/submodules/simple-knn/simple_knn.cu b/submodules/simple-knn/simple_knn.cu index 661d488..0d0329c 100755 --- a/submodules/simple-knn/simple_knn.cu +++ b/submodules/simple-knn/simple_knn.cu @@ -20,6 +20,7 @@ #include #include #include +#include #define __CUDACC__ #include #include @@ -218,4 +219,4 @@ void SimpleKNN::knn(int P, float3* points, float* meanDists) boxMeanDist << > > (P, points, indices_sorted.data().get(), boxes.data().get(), meanDists); cudaFree(result); -} \ No newline at end of file +}