当前位置:   article > 正文

Linux服务器安装pytorch3d完整流程_linux安装pytorch3d

linux安装pytorch3d

接上一篇安装tiny-cuda-nn完整流程,本人现在记录一下安装pytorch3d完整流程,同时供各位一块安装,省去一些不必要的时间。

首先需要提醒的是,虽然其他博客提到直接使用conda install和pip install十分的方便,但是由于这两种安装的包的编译环境大概率和你自己的虚拟环境不一样,会导致安装的包无法使用,所以强烈建议从git上自己手动进行安装!!

1、命令行安装(最方便快捷)

直接从github上面自动进行pip安装:

pip install git+https://github.com/facebookresearch/pytorch3d.git

同时再次提醒各位,如果发生服务器连接不上github的情况,建议直接换成gitee进行安装,此处放出我自己clone的仓库链接:

https://gitee.com/richard_chenRuiyang/pytorch3d

同时本人想提醒各位,如果从github上面clone包失败,可以先将包迁移到自己的gitee上,亲测十分有效,可以减少很多时间,具体可以参考如下文章:

【经验】Git|Linux终端git太慢,改hosts、复制文件夹、用镜像源?不不不不不

2、手动安装(稍微麻烦)

为什么我的两篇文章中都会提到使用手动安装,是因为在我的服务器上,build wheel总是失败,总是需要一个pyproject,所以我的安装都是使用手动的方法的。

其次还需要说明,根据其他博客,pytorch3d源码是安装不成功的,需要使用修改后的代码,经实测修改后的包没有问题,https://gitee.com/richard_chenRuiyang/pytorch3d链接中的包是修改后的包,可以直接使用。

(1)将pytorch3d包git clone到虚拟环境中:

git clone https://gitee.com/richard_chenRuiyang/pytorch3d

(2)安装pytorch3d的依赖包:

  1. conda install fvcore iopath
  2. conda install -c bottler nvidiacub

(3)确认系统的gcc和g++版本

安装pytorch3d包的时候gcc和g++版本需要是7.x,如果不符合要求在网上很容易搜到gcc,g++版本切换的方法,在此不过多赘述。

(4)修改setup.py文件

在我安装包的时候,总是出现一些关于ninja的问题,在参考了这篇文章之后找到了解决方法:​​​​​​出现错误“subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1”解决方法

所以在此步,我们需要打开pytorch3d/setup.py文件,定位到最后一行,将:

cmdclass={'build_ext': BuildExtension}

改为:

cmdclass={'build_ext': BuildExtension.with_options(use_ninja=False)}

这样在之后的安装过程中就不会出现问题了。

(5)安装pytorch3d包

使用以下指令即可正确安装pytorch3d包:

python setup.py install

(6)检查是否安装成功

在安装结束后可以通过pip list等指令检查是否成功安装包。

在我的安装过程中,遇到了很多奇奇怪怪的问题,花了两天时间最终成功装好,在此感谢csdn各位大佬的博客,也希望各位能够通过我的博客很快安装好pytorch3d包。在此引用其他博客的一句话,在安装pytorch3d的时候不要一直使用一个方法,一个方法进行不下去的时候迅速更换其他方法进行安装。

在此非常感谢csdn的前辈,本篇博客参考以下文章:

ubuntu安装pytorch3d

​​​​​​出现错误“subprocess.CalledProcessError: Command ‘[‘ninja‘, ‘-v‘]‘ returned non-zero exit status 1”解决方法

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/IT小白/article/detail/588315
推荐阅读
相关标签
  

闽ICP备14008679号