安装PyTorch,装pytorch的方法

  安装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的观点,版权归原作者所有,仅为传播更多信息之目的,如有侵权请联系,我们将第一时间修改或删除,多谢。

留言与评论(共有 条评论)
   
验证码: