PyTorch中apex安装方式和避免踩坑
更新时间：2022年07月05日 09:27:30 作者：大梦冲冲冲
本文主要介绍了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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
这篇文章主要介绍了python__name__原理及用法详解,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2019-11-11
这篇文章主要介绍了Django 表单模型选择框如何使用分组,小编觉得挺不错的，现在分享给大家，也给大家做个参考。一起跟随小编过来看看吧2019-05-05
今天小编就为大家分享一篇Python+selenium点击网页上指定坐标的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-07-07
本文详细讲解了Python并发编程之IO模型，文中通过示例代码介绍的非常详细。对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下2022-06-06
这篇文章主要介绍了Virtualenv 搭建 Py项目运行环境的详细教程，本文通过图文并茂的形式给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值,需要的朋友可以参考下2020-06-06
Python+unittest+requests 接口自动化测试框架搭建教程
这篇文章主要介绍了Python+unittest+requests 接口自动化测试框架搭建教程，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2020-10-10
Python和Rust都采用了垃圾收集（Garbage Collection）机制来管理内存,但它们各自的实现方式有很大的不同,下面就跟随小编一起来深入了解下二者的区别吧2024-03-03
这篇文章主要介绍了python实现rest请求api示例,需要的朋友可以参考下2014-04-04
这篇文章主要给大家介绍了关于pycharm打开chrome自动退出错误的解决方法,文中通过实例代码以及图文将解决的办法介绍的非常详细,需要的朋友可以参考下2023-07-07
今天小编就为大家分享一篇python 从文件夹抽取图片另存的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-12-12
最新评论