search

Sense-X / TSD

1st place models in Google OpenImage Detection Challenge 2019
Apache License 2.0
456 stars 64 forks source link

How to use it with A100 Gpus? #37

Closed HenryHZY closed 2 years ago

HenryHZY commented 2 years ago

I have tried four environments to use it it with A100, but failed.

(1)
a100
torch1.4.0+cudatoolkit10.0
nccl2+cuda10.0
mmcv 0.4.4
gcc7
ERROR
(2)
a100
torch1.4.0+cudatoolkit10.1
nccl2+cuda10.1
mmcv 0.4.4
gcc7
ERROR
(3)
dockerfile that you provided
mmcv-0.4.4
(4)
py3.6 + torch1.7 + cuda11.0 + gcc7 + nccl2.7.3

I really spend a lot of time in the environment setting. But it seems that TSD is out-of-dated currently: A100 needs cuda11.0 and pytorch>=1.7 TSD needs mmcv0.4.4 + mmdet1.0 + pytorch<=1.4

Could anyone offer a TSD + mmdet2.x code with me? Thanks a lot!