安装PyTorch,装pytorch的方法
本文主要介绍PyTorch中apex的安装方法以及如何避免踩坑。通过示例代码进行了非常详细的介绍,对于大家的学习或者工作都有一定的参考价值。有需要的朋友下面跟边肖学习。
00-10101,apex2,安装步骤3,虽然只是三行简单的命令,但是后台还是有很多“漏洞”:这个库的安装并不是像其他人一样直接以pip的形式安装XXX,而是以原有的Git方式。
目录
这是NVIDIA开发的基于PyTorch的混合精准训练加速神器,可以提高运算速度,减少内存占用。
1、apex
git克隆https://github.com/NVIDIA/apex.git
cd顶点
pip install-v-no-cache-dir-global-option=-CPP _ ext -global-option=-cuda _ ext 。/
2、安装步骤
(1)git clone克隆失败——3354解决方案:直接从网页下载。
如果出现这种情况,那么我们会选择“曲线救国”。直接在GitHub下载库,然后解压。并放在相应的位置。
(2)安装步骤中的第三行命令出错。
错误内容:运行时错误: cuda扩展正在使用与用于编译py torch二进制文件的版本不匹配的cuda版本进行编译。py torch二进制文件是用cuda 9.0.176编译的。
解决方法:错误的意思是cuda和pytorch的版本不对应,但是通过查找可以不用- global - option使用。
然后,修改第三行命令,如下所示:
pip安装-v - no-cache-dir。/
最后希望大家安装成功。走吧。
关于如何在PyTorch中安装apex以及如何避免踩坑的这篇文章到此为止。有关PyTorch apex安装的更多信息,请搜索热门IT软件开发工作室之前的文章或继续浏览下面的相关文章。希望大家以后多多支持热门IT软件开发工作室!
郑重声明:本文由网友发布,不代表盛行IT的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。