search

MILVLG / bottom-up-attention.pytorch

A PyTorch reimplementation of bottom-up-attention models
Apache License 2.0
294 stars 76 forks source link

can you tell me the apex version suitable for torch 1.4? #109

Open qq123aa456 opened 1 year ago

qq123aa456 commented 1 year ago

I met a trouble.when I use torch>1.4 I can't sucessful install detectron2 due to subprocess.CalledProcessError: Command '['ninja', '-v']' returned non-zero exit status 1. but when I use torch 1.4 I successsful install detectron2 but can extract features beacause an error module 'torch.cuda' has no attribute 'amp'. is there anyone who sucssful extract features? I really need help.

qq123aa456 commented 1 year ago

I search the issue they told me amp appear pytorch>1.6 ,which is so bad

niuzhustc commented 1 year ago

I search the issue they told me amp appear pytorch>1.6 ,which is so bad

I use torch 1.5+cu 101+cp36 on 2080ti, and my apex version is 22.04, I use the detectron2 v0.3, maybe you could try this.