安装pytorch3d最简单方法

这篇具有很好参考价值的文章主要介绍了安装pytorch3d最简单方法。希望对大家有所帮助。如果存在错误或未考虑完全的地方,请大家不吝赐教,您也可以点击"举报违法"按钮提交疑问。


前言

安装pytorch3d踩了很多坑,现将最简单的方法公布如下:


一、pytorch3d是什么?

PyTorch3D的目标是帮助加速深度学习和3D交叉点的研究。3D数据比2D图像更复杂,在从事Mesh R-CNN和C3DPO等项目时,我们遇到了一些挑战,包括3D数据表示、批处理和速度。我们开发了许多有用的算子和抽象,用于3D深度学习,并希望与社区分享,以推动这一领域的新研究。
在PyTorch3D中,我们包含了高效的3D操作符、异构批处理功能和模块化可微渲染API,为该领域的研究人员提供了急需的工具包,以实现复杂3D输入的前沿研究。

来自:https://pytorch3d.org/docs/why_pytorch3d

二、安装步骤

1.添加anaconda源(最最最最最关键!!)

以清华源 ubuntu18.04为例,其他平台安装方法可以借鉴:

# 在终端执行以下代码
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
# 以上两条是Anaconda官方库的镜像

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
# 以上是Anaconda第三方库 Conda Forge的镜像

# for linux
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
# for legacy win-64
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
#以上两条是Pytorch的Anaconda第三方镜像
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch3d/

conda config --set show_channel_urls yes

添加完成后可以使用conda info命令查看是否添加成功
!!!添加pytorch源和pytorch3d源是最关键的步骤!!!

2.创建环境

以python3.8为例,这个看实际需求。

conda create -n pytorch3d python=3.8
conda activate pytorch3d

3.安装pytorch和pytorch3d

截止本文的编写时间,当前最版本pytorch3d为0.6.2,最高支持pytorch1.11.0,所以最好不要安装pytorch 1.12.0 以上版本;以pytorch 1.10.1版本为例,对了我的nvcc -v cuda版本是11.3,安装pytorch需要适配:


# 首先安装pytorch
conda install pytorch==1.10.1 torchvision==0.11.2 torchaudio==0.10.1 cudatoolkit=11.3
# 使用官方网站提供的命令安装,参考:https://pytorch.org/get-started/previous-versions/
# 但是尤其注意以因为已经更换过conda源了,所以不需要添加-c pytorch -c conda-forge,否则默认从官方源添加
# 这一步需要等待一段时间,但清华源速度总比官方源强多了

# 其次安装必要库
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

# 安装pytorch3d,只需要一条命令
conda install pytorch3d

那么就是激动人心的时刻了: 4.更新一下编译安装吧

安装完自己的所需的其他环境后,PyTorch、fvcore、iopath是必须的。

编译安装实际上是最不容易出错的:
参考官网安装教程 https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md

  1. Install from GitHub 如果可以科学上网,执行这条
pip install "git+https://github.com/facebookresearch/pytorch3d.git"

如果要使用发布版本的代码进行安装,而不是从主分支中获取,请使用以下方法。

pip install "git+https://github.com/facebookresearch/pytorch3d.git@stable"

对于版本早于CUDA 11的CUDA构建,如上所述,在构建之前按照官方文档设置CUB_HOME。

在macOS上从Github安装:应该提供一些环境变量,比如这样。

MACOSX_DEPLOYMENT_TARGET=10.14 CC=clang CXX=clang++ pip install "git+https://github.com/facebookresearch/pytorch3d.git"
  1. Install from a local clone 本地安装适用于大部分情况,可以下载zip之后解压
git clone https://github.com/facebookresearch/pytorch3d.git
cd pytorch3d && pip install -e .

要在从本地克隆运行安装后重建,请rm-rf build/**/*.然后pip install -e。在重新安装PyTorch之后,您通常需要重建pytorch3d。对于版本早于CUDA 11的CUDA构建,如上所述,在构建之前设置CUB_HOME。

从macOS上的本地克隆安装:

MACOSX_DEPLOYMENT_TARGET=10.14 CC=clang CXX=clang++ pip install -e .

从Windows上的本地克隆安装:根据PyTorch的版本,在编译之前可能需要更改某些PyTorch标头。这些问题经常在此存储库中的问题中进行讨论。在进行任何必要的修补后,您可以转到“x64 Native Tools命令提示符for VS 2019”进行编译和安装

cd pytorch3d
python3 setup.py install

安装后,您可以运行单元测试

python3 -m unittest discover -v -s tests -t .

总结

完结撒花,希望后续在使用pytorch3d的过程中能够出一些好的教程,期待自己可以~

有问题请留言, 成功的童鞋麻烦留言告知啦。文章来源地址https://www.toymoban.com/news/detail-406213.html

到了这里,关于安装pytorch3d最简单方法的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!

本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处: 如若内容造成侵权/违法违规/事实不符,请点击违法举报进行投诉反馈,一经查实,立即删除!

领支付宝红包 赞助服务器费用

相关文章

  • pytorch3d成功安装

    一、pytorch3d是什么? PyTorch3D的目标是帮助加速深度学习和3D交叉点的研究。3D数据比2D图像更复杂,在从事Mesh R-CNN和C3DPO等项目时,我们遇到了一些挑战,包括3D数据表示、批处理和速度。我们开发了许多有用的算子和抽象,用于3D深度学习,并希望与社区分享,以推动这一领

    2024年02月12日
    浏览(41)
  • PyTorch3D 安装记录

    PyTorch 官网:PyTorch 官网 最新版本的 PyTorch:最新版本的 PyTorch 以 cuda 11.3,Python 3.9,Pytorch 1.12.1 为例 直接使用 pip 安装 PyTorch3D 会提示错误 进入 Anaconda 官网找到 PyTorch3D :PyTorch3D Anaconda 官网 找到和 cuda 以及 Python 版本对应的 Pytorch3D 版本 ,找到正确的版本后,右键绿色的文件

    2024年02月14日
    浏览(47)
  • Windows安装Pytorch3d

    安装Visual Studio 2019 【我记得必须是2017-2019之间的版本,我一开始用的是2022的版本就安装不了】网址 pytorch和pytorch3d、cuda和NVIDIA CUB版本需要相互对应 pytorch和pytorch3d版本对应关系如下:https://github.com/facebookresearch/pytorch3d/releases cuda和NVIDIA CUB版本对应关系如下:https://github.com/

    2024年02月09日
    浏览(40)
  • Pytorch3D安装全流程-亲自安装

    简单介绍 Pytorch3D是一个用于加速深度学习在处理3D相关信息时候的运算速率的库。深度学习有时会需要处理大量的3D数据,比如在人脸建模的时候。这时候使用这些库会大大帮助我们节省运算成本。 遇到问题 按照官方的安装方法会遇到不少问题:https://github.com/facebookresearch

    2024年01月17日
    浏览(38)
  • linux 安装pytorch3d的坑

    事实上,只要按照官方文档的说明就可以完美安装。其中坑的地方在于conda的管理可能会导致下载的版本 不符合你的要求 (例如下载成了cpu版本、下载的cuda版本)而同样尝试使用源码编译以及其他方式下载库都会导致同样的问题,这里主要的原因是由于python的版本不对以及

    2024年02月16日
    浏览(43)
  • Pytorch3D Linux环境下安装(踩坑)记录

    Pytorch3D Linux环境下安装(踩坑)记录 文章目录 前言 一、准备工作 二、极简安装(Installing prebuilt binaries) 1.创建环境 2.安装Pytorch 3.安装两个核心库 4.(非必要)Tests/Linting and Demos 相关 5.安装Pytorch3D 三、手动安装(Building from source) 1.创建环境 2.安装Pytorch 3.安装两个核心库(

    2024年02月13日
    浏览(39)
  • pytorch3d安装遇到的一些坑和解决过程

    最近为了安装pytorch3d,折腾了两天,足足两天! 要注意,安装pytorch3d之前,必须先安装pytorch torchvision Pytorch3d官方 是建议用conda install的方式来安装,也可以用pip来安装: pip install PyTorch torchvision -c pytorch-nightly 1.首先说一下MacOS M1 芯片安装情况,刚开始使用的是: 会安装完成

    2024年02月06日
    浏览(37)
  • Windows 11 安装 pytorch3d可能遇到的问题,以及最终成功安装使用的版本分享(使用RTX3070)

    Win10下pytorch3D安装方法   由于pytorch3d对于pytorch、CUDA、CUB的版本对应实在是过于严苛,所以我的建议是直接找别人成功安装的软件版本列表对着抄,pytorch啥的该重装重装,反正可以用虚拟环境,CUDA也是,一台电脑共存两个版本的CUDA也不是不行。唯一麻烦一点的可能就是G

    2024年02月20日
    浏览(45)
  • win下配置pytorch3d

    你可能觉得pytorch3d 0.6.0版本有点低,但是折腾不如先配上用了,以后有需要再说。 (后话:py 3.9 + pytorch 1.12.1 + cuda 11.3_cudnn 8_0 + pytorch3d 0.7.1 + CUB 1.11.0 也OK的) 1.1 创建新环境 1.2 找之前的pytorch版本,我这里直接给出我的选择的版本 一行行复制粘贴运行下面的命令 1.3 下载对应

    2024年02月08日
    浏览(38)
  • Pytorch3d中的倒角损失函数Chamfer Distance Loss的用法(pytorch3d.loss.chamfer_distance)

    API文档在这里 源码在这里 之前看到的一个干货满满的Pytorch3D安装指导与简单例子的帖子在这里  官方tutorials中提到的用法在下面这个代码块里面,前面后面东西都挺多的就把和chamfer_distance相关的摘到了这里

    2024年02月15日
    浏览(49)

觉得文章有用就打赏一下文章作者

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

请作者喝杯咖啡吧~博客赞助

支付宝扫一扫领取红包,优惠每天领

二维码1

领取红包

二维码2

领红包