美文网首页
pytorch3d 配置

pytorch3d 配置

作者: 去大海边吹海风 | 来源:发表于2021-06-25 17:40 被阅读0次

    尝试windows下,linux下pytorch3d的安装,各种不成功,每次我都不是幸运的那个一次成功,这样有幸记录下自己的坎坷,希望大家可以节省时间在配置上。

    windows

    一、pip install pytorch3d 可以成功,但是是0.01版本,太老了,库比较少,所以基本淘汰了。

    二、在github下载的v0.2.0版本
    python3.6.8 + cuda10.1+ cudnn7 + pytorch1.7.1+cu101 (其他版本应该影响不大)

    1. 下载pytorch3d 到本地,下载release版本 https://github.com/facebookresearch/pytorch3d/archive/v0.2.0.zip

    2. 以管理员方式打开x64 Native Tools Command Prompt for VS 2019

    3. 进入pytorch3d目录
      cd pytorch3d-0.2.0
      python setup.py install

    出现以下语句即为安装成功,然而0.2.0包还不完善
    Using d:\programs\python3.6.8\lib\site-packages
    Finished processing dependencies for pytorch3d==0.2.0

    ————————————————————————————————————————
    也许你会遇到下面的问题
    File "D:\Programs\python3.6.8\lib\site-packages\torch\utils\cpp_extension.py", line 1529, in _run_ninja_build
    raise RuntimeError(message)
    RuntimeError: Error compiling objects for extension

    按照这里面的 https://github.com/facebookresearch/pytorch3d/issues/323

    同样方式0.4.0暂时没有成功,待更新

    Linux

    一、作者已经编译了python3.6,3.8 和 pytorch171,170版本的库,你可以修改版本试试看看存在不存在
    https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py36_cu101_pyt170/download.html
    pip install ./pytorch3d-0.4.0-cp36-cp36-linux_x86_64.whl

    二、自己在python3.6,+pytorch1.5.0+cu101上没有自己编译成功,如果有编译成功的同胞可以分享一下。

    引用:
    https://www.icode9.com/content-3-1020527.html
    https://github.com/facebookresearch/pytorch3d/issues/717

    相关文章

      网友评论

          本文标题:pytorch3d 配置

          本文链接:https://www.haomeiwen.com/subject/nmugyltx.html