Environment for pytorch1.9 version

[YuxuanChen21]

when running

python setup.py develop 

in pytorch1.9 version using docker, with environment as python==3.8, torch==1.9.0 and cudatoolkit==11.0,
there exist error:
/opt/conda/envs/s2anet/compiler_compat/ld: cannot find -lc10_cuda
/opt/conda/envs/s2anet/compiler_compat/ld: cannot find -ltorch_cuda
collect2: error: ld returned 1 exit status
error: command 'g++' failed with exit status 1

I wonder if it is due to environment differences. Could you please give us some suggestions? Thank you!

[csuhan]

open-mmlab/mmdetection#2430 may help you.