STTran icon indicating copy to clipboard operation
STTran copied to clipboard

fasterRCNN compilation error with python 3.8 and pytorch 1.12 cuda 11.3

Open zee-fee opened this issue 2 years ago • 2 comments

I.m trying to setup sttran from scratch and I'm getting errors while setting up fasterRCNN package. The command I run is : STTran/fasterRCNN/lib: python setup.py install

fatal error: THC/THC.h: No such file or directory
    5 | #include <THC/THC.h>
      |          ^~~~~~~~~~~
compilation terminated.

zee-fee avatar Aug 05 '22 14:08 zee-fee

The CUDA version you are using is too high. If you are not using a RTX30 GPU, it is better to downgrade to Cuda10.2.

yrcong avatar Aug 10 '22 09:08 yrcong

hi zee-fee, you can download the fix here https://github.com/jwyang/faster-rcnn.pytorch/pull/894

ScottHoang avatar Aug 11 '22 18:08 ScottHoang

I downgraded to cuda 10.2 and now it works with torch 1.1

zee-fee avatar Aug 18 '22 20:08 zee-fee