fasterRCNN compilation error with python 3.8 and pytorch 1.12 cuda 11.3

yrcong / STTran

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021

MIT License

188 stars 34 forks source link

fasterRCNN compilation error with python 3.8 and pytorch 1.12 cuda 11.3 #36

Closed zee-fee closed 2 years ago

zee-fee commented 2 years ago

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.

yrcong commented 2 years ago

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.

ScottHoang commented 2 years ago

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

zee-fee commented 2 years ago

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