【安装部署】Linux下最简单的 pytorch3d 安装-CSDN博客

本文链接：https://blog.csdn.net/qq_43426908/article/details/147427106

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

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

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 放到服务器上；

执行安装命令：

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 下测试以下代码：

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

安装成功！

在这里插入图片描述

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