> 技术文档 > 【安装部署】Linux下最简单的 pytorch3d 安装_pytorch3d安装

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

  最近接触一个项目需要用到 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

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

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 # 安装

  安装完成显示如下:

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

4、验证安装成功

  python 下测试以下代码:

from pytorch3d import _Cfrom pytorch3d.structures import Meshes, Pointclouds

  安装成功!

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


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