linux安装apex时的一些问题
2021/5/30 7:26:15
本文主要是介绍linux安装apex时的一些问题,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
简要描述:参考apex 安装/使用 记录 的方法安装,运行代码后遇到问题。适合当下(2021.5.29)的安装方法见最后。
问题1
ModuleNotFoundError: No module named 'fused_adam_cuda'
解决
apex的旧版本较老,参考 https://github.com/NVIDIA/apex/issues/633 这个问题中有一个回答:
you can try FusedAdam in apex.optimizers instead of in apex.optimizers.contrib
主要是因为apex的版本更迭,引入FusedAdam的包名发生了变化。
参考 apex关于FusedAdam的官方文档,把
from apex.contrib.optimizers import FusedAdam
改成
from apex.optimizers import FusedAdam
此外,如果使用 FP16_Optimizer
, 引入的包名也发生了变化。
参考 apex关于FP16_Optimizer的官方文档,把
from apex.contrib.optimizers import FP16_Optimizer
改成
from apex.fp16_utils import FP16_Optimizer
问题2
ModuleNotFoundError: No module named 'fused_layer_norm_cuda'
解决
参考ModuleNotFoundError: No module named ‘fused_layer_norm_cuda’。之前安装apex出现问题,卸载后,进入git后得到的apex文件夹重装。
pip uninstall apex python setup.py install --cpp_ext --cuda_ext
安装
参考ModuleNotFoundError: No module named ‘fused_layer_norm_cuda’,适合当下(2021.5.29)的安装方法为:
git clone https://github.com/NVIDIA/apex cd apex python setup.py install --cpp_ext --cuda_ext
这篇关于linux安装apex时的一些问题的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-06-0600-macOS和Linux安装和管理多个Python版本
- 2024-03-30[译]漫画SELinux概念
- 2024-03-29linux 移动文件
- 2024-03-28linux .so file
- 2024-03-28Linux 磁盘管理
- 2024-03-28Linux学习笔记(十三)磁盘管理(一):磁盘分区
- 2024-03-26linux 创建 文件
- 2024-03-25使用SecureCRT对Linux vim进行颜色设置
- 2024-03-202019-2020-12 20199317 《Linux内核原理与分析》 第十二周作业
- 2024-03-20Linux运维的第二周总结