PyTorch中apex安装方式和避免踩坑
背景：这个库的安装不是像其他的一样的直接使用 pip install XXX的形式，而是使用原始的Git方式
1、apex
这是NVIDIA开发的基于PyTorch的混合精度训练加速神器，能够增加运算速度，并且减少显存的占用
2、安装步骤
git clone https://github.com/NVIDIA/apex.git
cd apex
pip install -v --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./3、虽然就简单三行命令，但是仍有很多“坑”存在
（1）git clone 克隆失败————解决方式：直接通过网页下载
如果出现这种情况，那我们就选择“曲线救国”。在GitHub中直接下载该库，然后解压。并放到相应的位置上。
（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 ./
最后的最后，希望大家都能成功安装，冲冲冲！
到此这篇关于PyTorch中apex安装方式和避免踩坑的文章就介绍到这了,更多相关PyTorch apex安装内容请搜索编程网以前的文章或继续浏览下面的相关文章希望大家以后多多支持编程网！
免责声明：
① 本站未注明“稿件来源”的信息均来自网络整理。其文字、图片和音视频稿件的所属权归原作者所有。本站收集整理出于非商业性的教育和科研之目的，并不意味着本站赞同其观点或证实其内容的真实性。仅作为临时的测试数据，供内部测试之用。本站并未授权任何人以任何方式主动获取本站任何信息。
② 本站未注明“稿件来源”的临时测试数据将在测试完成后最终做删除处理。有问题或投稿请发送至: 邮箱/279061341@qq.com QQ/279061341
