ViG CUDA and Python Version

[Stefan Scholbe]

Could you please specify the CUDA, apex, and Python versions you used to train ViG? Thank you.

[Kai Han]

pytorch1.7, cuda10.2, python 3.7, apex0.04

[ZeroerWiser]

pytorch1.7, cuda10.2, python 3.7, apex0.04

请问apex的版本是否可以是任意的呢？我看我现在安装的apex过程中称torch1.7版本过旧，即

[Kai Han]

apex的版本不能任意，最好下载apex0.04版的

[janesunflower]

apex的版本不能任意，最好下载apex0.04版的

请问apex0.04是哪个链接？

[Nur Muhammad Himawan]

can you put official link apex 0.04?

[Kai Han]

can you put official link apex 0.04?

https://github.com/huawei-noah/Efficient-AI-Backbones/tree/c47afb01dcdc448a4743e1ec2afda152c175e3e5/wavemlp_pytorch/apex-master