We use NVIDIA Quadro P5000 GPU....We are not able to run build.py file of nms....so, may get error of ModuleNotFoundError: No module named 'nms._ext'....Would You please like to assist me that which architecture have I to define in following syntax....?
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=[arch]
We use NVIDIA Quadro P5000 GPU....We are not able to run build.py file of nms....so, may get error of ModuleNotFoundError: No module named 'nms._ext'....Would You please like to assist me that which architecture have I to define in following syntax....?
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=[arch]