参见windows 安装apex_Nvidia Apex安装
重点:
1、在安装前先检查一下,电脑的cuda版本和pytorch内的cuda版本是否一样,不一样的话就把低版本的进行升级;

$ git clone https://github.com/NVIDIA/apex

$ cd apex

2、在保证cuda版本一致的前提下进入apex文件夹,使用命令:python setup.py install 即可安装成功

更多推荐