ModuleNotFoundError: No module named 'apex'

 

git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

아래의 주소로 가서 참조하면 된다.

https://github.com/NVIDIA/apex

GitHub - NVIDIA/apex: A PyTorch Extension: Tools for easy mixed precision and distributed training in Pytorch