qywu/TorchFly

ModuleNotFoundError: No module named 'apex.normalization'

Closed this issue · 2 comments

luweishuang commented

import torchfly and get a error

qywu commented

You should install apex first with CUDA support.

# modified the error due to cuda version
git clone https://github.com/qywu/apex
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./
  • 👎2
netsafe commented

there's no such repository for now - where can I get the code?

  • 👀4