【安装部署】Linux下最简单的 pytorch3d 安装

风巽·剑染春水2025-04-23 12:15

最近接触一个项目需要用到 pytorch3d 库，找了好多眼花缭乱的资料，最后发现大道至简~

1、查看 python，cuda，pytorch 的版本

bash 复制代码

source activate myMRI   # 激活环境
python --version   # python 版本查看
nvcc --version     # cuda 版本查看
python -c "import torch; print(torch.__version__)"   # pytorch 版本查看

输出如下，可见我的 python 版本是 3.8，cuda 版本是 11.7，pytorch 版本是 1.13.1

2、下载版本对应的 pytorch3d 包

官方下载地址：https://anaconda.org/pytorch3d/pytorch3d/files

下载与 python，cuda，pytorch 对应的版本：

3、安装 pytorch3d

将下载好的 pytorch3d-0.7.5-py38_cu117_pyt1131.tar.bz2 放到服务器上；

执行安装命令：

bash 复制代码

source activate myMRI   # 激活环境
cd /home/qijing/anaconda3/envs  # 进入pytorch3d-0.7.5-py38_cu117_pyt1131.tar.bz2所在路径
conda install pytorch3d-0.7.5-py38_cu117_pyt1131.tar.bz2  # 安装

安装完成显示如下：

4、验证安装成功

python 下测试以下代码：

python 复制代码

from pytorch3d import _C
from pytorch3d.structures import Meshes, Pointclouds

安装成功！

技能up，٩(๑•̀ω•́๑)۶