Pytorch3D安装全流程-亲自安装

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

简单介绍

Pytorch3D是一个用于加速深度学习在处理3D相关信息时候的运算速率的库。深度学习有时会需要处理大量的3D数据,比如在人脸建模的时候。这时候使用这些库会大大帮助我们节省运算成本。

遇到问题

按照官方的安装方法会遇到不少问题:https://github.com/facebookresearch/pytorch3d/blob/main/INSTALL.md。我个人在安装的时候,输入相关的指令并不能够成功安装,也是折腾了快一个上午加一个中午才解决,这里分享一下,希望能帮助更多的人。

准备工作

1.conda创建好的虚拟环境

首先你需要一个用conda创建好的虚拟环境,如何创建网上教程很多,在此不多赘述。注意你的cuda里面要有cudatoolkit这个包,这个包是用来使用cuda的,简单来说就是调用你电脑GPU的。

要记好你虚拟环境的python/pytorch版本和cudatoolkit版本,建议pytorch版本手动设定安装版本,否则会自动安装到最新版,最新版往往与别的很多包都是不兼容的

2.cub安装配置

cub的版本一定要和cudatoolkit版本对应,版本对应说明的链接在这里:https://gitcode.net/mirrors/nvidia/cub?utm_source=csdn_github_accelerator

知道自己版本是什么了之后,开始下载,下载链接:https://github.com/NVIDIA/cub/releases

pytorch3d安装,# Pytorch installation,pytorch,python,人工智能

进入网页后往下翻,找到对应版本的Assets。这里有两个压缩包,一般如果是windows电脑的话选择zip,如果是linux的话选择tar.gz(然后用命令解压)

之后解压相应安装包。

解压完成后,将文件所在位置目录添加到环境变量中

环境变量设置方式:右键我的电脑(此电脑)—属性—高级系统设置—环境变量—Path变量——添加新环境变量

3.MinGW安装

网上有博客说要安装,因为我电脑里之前需要用vscode所以安装过,电脑里就有,所以我也不确定是否一定要装。

正式安装

1.首先下载pytorch3D安装包

下载链接:https://github.com/facebookresearch/pytorch3d/releases

一定要注意版本对应!!

pytorch3d安装,# Pytorch installation,pytorch,python,人工智能

比如上面这张图,就说明pytorch3D 0.7.1 版本只支持1.9.0到1.12.1的pytorch

下载后进行解压,建议安装到site-packages这个文件夹里,因为这个文件夹是专门存放pytorch各种包的地方,比如我的:.\anaconda3\Lib\site-packages

2.更改相关的文件

解压好的文件并不能直接使用,需要先改一下代码。

打开pytorch3D文件夹(我的文件名是pytorch3d-0.7.2),里面有个setup.py文件夹,修改下面部分的代码:

pytorch3d安装,# Pytorch installation,pytorch,python,人工智能

52行我注释掉的是原来的代码,53行有红点的是修改后的。我猜这里修改是不指定C++的编译器。因为其实python的底层代码还是C++,这个包下载好并不能直接用,需要安装编译后这个包才能用。

网上有的教程改了很多地方的代码,但是我自己只修改了这一个地方就可以使用了。如果只改了一个没有成功,参考这篇博客(点击链接)

3.安装VS2019

需要安装vs2019,这里上网找下教程即可,教程很多。下载vs2019的原因是用来编译下载的安装包。

等下载安装好之后,找到这玩意儿x64 Native Tools Command Prompt for VS 2019(如果是win10win11可以直接在搜索栏里搜索)。这个是vs2019自带的终端。

4.安装!!!

终于到这一步了。

首先依次在终端窗口输入下面的命令:

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

然后再输入这个对pytorch3d进行编译安装:

python setup.py install

我记得当时终端里跑了好多好多行代码,应该是在一行行编译,大概过了个十五分钟,就成功了。文章来源地址https://www.toymoban.com/news/detail-798977.html

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

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

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

相关文章

  • linux 安装pytorch3d的坑

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

    2024年02月16日
    浏览(28)
  • 安装pytorch3d最简单方法

    安装pytorch3d踩了很多坑,现将最简单的方法公布如下: PyTorch3D的目标是帮助加速深度学习和3D交叉点的研究。3D数据比2D图像更复杂,在从事Mesh R-CNN和C3DPO等项目时,我们遇到了一些挑战,包括3D数据表示、批处理和速度。我们开发了许多有用的算子和抽象,用于3D深度学习,

    2023年04月09日
    浏览(29)
  • Windows下Pytorch3d的安装方法

    一、问题: 近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为: PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch. 可知Pytorch3d库可以提供多种应用于3D视觉处理的组件,其在ubuntu下安装较为简单,但win下稍繁琐,本贴是对个人安

    2024年02月02日
    浏览(34)
  • 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日
    浏览(29)
  • Windows GPU 下Pytorch3d的安装方法

    我的笔记本是windows GPU 安装Pytorch3d失败,废了九牛二虎之力才安装好,为了纪念我的坚持我分享给大家。 Pytorch3d在WIN下的安装 一、问题: 近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为: PyTorch3D provides efficient, reusable components for 3D Computer Vision r

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

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

    2024年02月06日
    浏览(28)
  • Win10下pytorch3D安装方法,本人亲测可用

    本人的电脑配置如下: 目录 1.版本确认 2.下载cub 3.下载pytorch3d  4.下载VSC2019  5.编译pytorch3d源文件         这个步骤非常非常重要,因为不同的pytorch3d版本适配于不同的torch版本,你可以进到facebook/pytorch3d的官网查看不同的适配版本(这是0.7.1的),其余的你们可以自己查看

    2024年02月15日
    浏览(33)
  • 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日
    浏览(26)
  • Pytorch3d中的倒角损失函数Chamfer Distance Loss的用法(pytorch3d.loss.chamfer_distance)

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

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

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

    2024年02月20日
    浏览(32)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包