Linux下安装apex的过程及其遇到的问题

1.尝试pip install apex

先直接在终端窗口pip install apex安装

遇到的问题

  1. 需要安装pytoml>0.1和cryptacular
  2. 安装pytoml没问题,但是安装cryptacular需要更新enscons、attrs、editables、SCons等
  3. 安装完成之后,仍然无法安装cryptacular

直接通过github下载安装

github安装apex遇到问题及解决方法

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,所以需要切换环境变量

Pytorch 使用不同版本的cuda的方法

问题2

报错: error: command ‘gcc’ failed with exit status 1
解决: 输入命令git checkout f3a960f80244cf9e80558ab30f7f7e8cbf03c0a0
但是这样又出现问题:fatal: Not a git repository (or any of the parent directories): .git,我又重新从github克隆了apex,重新输入git命令,又报错:error: Your local changes to the following files would be overwritten by checkout:,解决方法:git切换分支,之后重新运行python命令安装apex,正常无问题。

版权声明:本文为CSDN博主「sumengmeng_」的原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/sumengmeng_/article/details/115026742

sumengmeng_

我还没有学会写个人说明!

暂无评论

发表评论

相关推荐

【论文复现】FPN

以RetinaNet里面的FPN为例,总共有5层,原始的FPN论文中只有P3,P4,P5三层。P6和P7是RetinaNet论文里面特有的。其中P6有两种实现方式,可以由P5

YOLO跌倒检测笔记

YOLO跌倒检测笔记 代码来源: https://github.com/qiaoguan/Fall-detection/blob/master/demo.gif 介绍: 上面原始代码是python2/open