apex——安装方式和避免踩坑

背景:这个库的安装不是像其他的一样的直接使用 pip install XXX的形式,而是使用原始的Git方式

1、apex

这是NVIDIA开发的基于PyTorch的混合精度训练加速神器,能够增加运算速度,并且减少显存的占用

2、安装步骤

1、git clone https://github.com/NVIDIA/apex.git
2、cd apex
3、pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

3、虽然就简单三行命令,但是仍有很多”坑”存在

(1)git clone 克隆失败————解决方式:直接通过网页下载

如果出现这种情况,那我们就选择 “曲线救国”。在GitHub中直接下载该库,然后解压。并放到相应的位置上。

apex——安装方式和避免踩坑

; (2)安装步骤中的第三行命令报错

报错内容:RuntimeError: Cuda extensions are being compiled with a version of Cuda that does not match the version used to compile Pytorch binaries.  Pytorch binaries were compiled with Cuda 9.0.176.

解决方式:其错误意思就是 cuda和pytorch的版本不对应,但是通过搜索也发现可以不带 –global –option 也能用
于是,修改第三行命令为:

pip install -v --no-cache-dir ./

最后的最后,希望大家都能成功安装,冲冲冲!

apex——安装方式和避免踩坑

Original: https://blog.csdn.net/qq_37844044/article/details/122625442
Author: 大梦冲冲冲
Title: apex——安装方式和避免踩坑

原创文章受到原创版权保护。转载请注明出处:https://www.johngo689.com/680710/

转载文章受原作者版权保护。转载请注明原作者出处!

(0)

大家都在看

亲爱的 Coder【最近整理,可免费获取】👉 最新必读书单  | 👏 面试题下载  | 🌎 免费的AI知识星球