1.尝试pip install apex
先直接在终端窗口pip install apex
安装
遇到的问题
- 需要安装pytoml>0.1和cryptacular
- 安装pytoml没问题,但是安装cryptacular需要更新enscons、attrs、editables、SCons等
- 安装完成之后,仍然无法安装cryptacular
直接通过github下载安装
git clone https://github.com/NVIDIA/apex.git
cd apex
python setup.py install --cpp_ext --cuda_ext
问题1
报错 :ModuleNotFoundError: No module named ‘fused_layer_norm_cuda‘
解决方法:命令使用的编译器是版本为cuda:10.0.130
,我的环境当前是cuda10.2
,所以需要切换环境变量
问题2
报错: error: command ‘gcc’ failed with