Relative Content

Tag Archive for pythoncmakecudasetup.py

Use CMake to build Cuda based python extension in setup.py

I’m using 3D Gaussian Splatting, which consists of a Cuda/C++ based python extension. I want to use CMake to compile it in setup.py. My basic idea is to generate .so file with CMake, and use this file to build python extension, which is similar to the idea in this question
However, it seems that my setup.py could not use this .so file, and failed to build my python extension. This is my setup.py and CMakeList. Can you help me?