ViG CUDA and Python Version

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

4.07k stars 708 forks source link

Open sscholbe opened 1 year ago

sscholbe commented 1 year ago

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

iamhankai commented 1 year ago

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

ZeroerWiser commented 1 year ago

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

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

iamhankai commented 1 year ago

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

janesunflower commented 1 year ago

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

请问apex0.04是哪个链接？

nurmuhimawann commented 1 year ago

can you put official link apex 0.04?

iamhankai commented 1 year ago

can you put official link apex 0.04?