Apex support from Lambda stack

Hello. Does anyone know how to install the APEX from Nvida?
I face some error to install this.
I usually use the Pytorch.
Do you have any idea for this?

Thank you.