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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家！
相关文章
plotly的Python软件包是一个开源的代码库，是一款擅长交互的Python绘图库。本文就来用plotly绘制正二十面体，需要的可以参考一下2023-02-02
这篇文章主要介绍了如何通过Python实现RabbitMQ延迟队列,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-11-11
这篇文章主要介绍了使用sklearn对多分类的每个类别进行指标评价操作，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2020-06-06
python写入csv时writerow()和writerows()函数简单示例
这篇文章主要给大家介绍了关于python写入csv时writerow()和writerows()函数的相关资料,writerows和writerow是Python中csv模块中的两个函数,用于将数据写入CSV文件,需要的朋友可以参考下2023-07-07
今天小编就为大家分享一篇PyCharm的设置方法和第一个Python程序的建立，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
pyttsx3是 Python 中的文本到语音的离线转换库,本文给大家介绍Python调用pyttsx3实现离线文字转语音的方式,感兴趣的朋友一起看看吧2024-03-03
python命令行解析之parse_known_args(）函数和parse_args()使用区别介绍
这篇文章主要介绍了python命令行解析之parse_known_args(）函数和parse_args()使用介绍,需要的朋友可以参考下2018-01-01
这篇文章主要介绍了python类方法中的self关键字使用,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教2023-12-12
基于tensorflow __init__、build 和call的使用小结
这篇文章主要介绍了基于tensorflow __init__、build 和call的使用小结，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2021-02-02
Python3调用百度AI识别图片中的文字功能示例【测试可用】
这篇文章主要介绍了Python3调用百度AI识别图片中的文字功能,结合实例形式分析了Python3安装及使用百度AI接口的相关操作技巧,并附带说明了百度官方AI平台的注册及接口调用操作方法,需要的朋友可以参考下2019-03-03
最新评论