本文主要记录在Linux上安装pytorch3d的步骤,其实按照官网教程就可以安装成功。
Install PyTorch3D


首先创建并进入一个新的环境:

conda create -n XXXX python=3.9
conda activate XXXX

然后依次输入以下命令:

conda install pytorch=1.13.0 torchvision pytorch-cuda=11.6 -c pytorch -c nvidia
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

conda install -c bottler nvidiacub

conda install pytorch3d -c pytorch3d

实测可以跑通。

注:上面的命令并不完整,但是对我来说已经足够,建议按官网的教程操作

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐