APEX是英伟达开源的,完美支持PyTorch框架,用于改变数据格式来减小模型显存占用的工具。
1、安装前电脑配置说明
(1)windows版本号:v11
(2)cuda版本号:11.0
(3)torch版本号:1.7.1+cu110
2、安装visual studio 2019
cuda v11.0只能匹配visual studio 2015-2019,我下载了2022版本后续编译时直接报版本不匹配。
3、下载apex源码
git clone https://github.com/NVIDIA/apex
4、apex安装
cd apex
python setup.py install即可。
ps:使用python setup.py install --cpp_ext --cuda_ext会报错如下所示: