I don't know what about you guys, buy using pip for instalin custom_rasterizer just won't work for me.
I have to manuallu define:
SET DISTUTILS_USE_SDK=1
SET TORCH_CUDA_ARCH_LIST=12.0+PTX
and then use .....\python_standalone\python.exe setup.py install
from the custome rasterizer folder for it to work!
Without doing this that way i get:
module 'custom_rasterizer' has no attribute 'rasterize'