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安装内容请搜索脚本之家以前的文章或继续浏览下面的相关文章希望大家以后多多支持脚本之家!
相关文章
2021年值得向Python开发者推荐的VS Code扩展插件
这篇文章主要介绍了2021年值得向Python开发者推荐的VS Code扩展插件,帮助大家更好的利用vscode进行python的开发,感兴趣的朋友可以了解下2021-01-01Python使用oslo.vmware管理ESXI虚拟机的示例参考
oslo.vmware是OpenStack通用框架中的一部分,主要用于实现对虚拟机的管理任务,借助oslo.vmware模块我们可以管理Vmware ESXI集群环境。2021-06-06python opencv实现直线检测并测出倾斜角度(附源码+注释)
这篇文章主要介绍了python opencv实现直线检测并测出倾斜角度(附源码+注释),文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧2020-12-12
最新评论