MIC-DKFZ / medicaldetectiontoolkit

The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Apache License 2.0
1.31k stars 297 forks source link

Issue building NMS for different GPU #107

Open kleingeo opened 4 years ago

kleingeo commented 4 years ago

I was trying to set-up the NMS one my comp (Windows 10) with a GTX 1070. I ran the command below, but got a cl.exe not found error. I then tried to run the script again, but instead of through a normal command prompt, I did it through the Developer Command Promt for VS 2019, which worked with the cl.exe not found error, but I then received a cluster of other errors. What is the approach to properly building the cuda objects? Going through other open and closed issues didn't provide any clear help. Does it have to do with the Cuda version installed?

Edit: I am using the Master branch. Would it better to switch to the update Torch1x branch?

cd cuda_functions/nms_3D/src/cuda/
nvcc -c -o nms_kernel.cu.o nms_kernel.cu -x cu -Xcompiler -fPIC -arch=sm_61