pytorch3d安装遇到的一些坑和解决过程

本文详细讲述了作者在MacOS M1和Ubuntu 18.04环境下安装PyTorch3d的曲折过程,包括conda和pip安装失败、源码安装的步骤与遇到的问题,以及版本兼容性的关键。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

最近为了安装pytorch3d,折腾了两天,足足两天!

要注意,安装pytorch3d之前,必须先安装pytorch torchvision

Pytorch3d官方 是建议用conda install的方式来安装,也可以用pip来安装:

pip install PyTorch torchvision -c pytorch-nightly

1.首先说一下MacOS M1 芯片安装情况,刚开始使用的是:

conda install pytorch3d -c pytorch3d

会安装完成,但是安装完以后执行.py文件跑程序时,会出现无法引用pytorch3d的问题,也尝试过用pip install pytorch3d ,这种方式会安装成最低版本:0.0.1,跑代码的时候依然出现无法应用的问题。

也尝试过这个方法:MacBook m1安装GPU版PyTorch与简单测试 - 知乎 并且修改了项目中的device值为‘mps’,如:device='mps',但是跑程序的时候还是会出现无法引用pytorch3d异常。搜索了好久,看到一个遇到同样的问题,最终通过源码安装成功了,加上刚开始我在ubuntu18.04 系统上安装了好久pytorch3d都没成功,然后我选择了源码安装

2.通过源码安装pytorch3d

git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d
pip install -e .
### 如何安装 PyTorch3D 为了成功安装 PyTorch3D,建议先设置好 Pythonconda 环境,并确保已经正确安装PyTorch 其他依赖项。以下是详细的安装指南: #### 创建 Conda 环境并激活 创建一个新的 conda 环境可以避免包冲突问题。使用如下命令来创建激活新的 conda 环境: ```bash conda create --name my_pytorch_env python=3.8 conda activate my_pytorch_env ``` #### 安装 PyTorch 及其依赖项 确认已按照官方文档中的说明选择了合适的 PyTorch 版本进行安装[^2]。 对于大多数用户来说,可以通过 pip 来快速安装最新版本的 PyTorch: ```bash pip install torch torchvision torchaudio ``` #### 安装 PyTorch3D PyTorch3D 支持通过 `pip` 或者源码编译两种方式进行安装。最简单的方式是利用预构建好的二进制文件来进行安装: ```bash pip install pytorch3d -f https://dl.fbaipublicfiles.com/pytorch3d/packaging/wheels/py38_cu111_pyt060/download.html ``` 注意 URL 中包含了特定于操作系统、CUDA 版本以及 PyTorch 版本的信息,请根据实际情况调整下载链接[^4]。 如果遇到任何兼容性问题,则可能需要考虑从源代码编译 PyTorch3D,但这通常不是必需的操作。 #### 验证安装是否成功 完成上述步骤之后,可以在 Python 解释器中运行下面这段简单的测试程序以验证 PyTorch3D 是否能够正常工作: ```python import pytorch3d print(pytorch3d.__version__) ```
评论 3
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值