Windows GPU 下Pytorch3d的安装方法

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

Windows下Pytorch3d的安装方法

我的笔记本是windows GPU 安装Pytorch3d失败,废了九牛二虎之力才安装好,为了纪念我的坚持我分享给大家。

Pytorch3d在WIN下的安装

一、问题:

近期在入手三维重建中,发现很多项目程序都需要Pytorch3d库,其官网定义为:

PyTorch3D provides efficient, reusable components for 3D Computer Vision research with PyTorch.

可知Pytorch3d库可以提供多种应用于3D视觉处理的组件,其在ubuntu下安装较为简单,但win下稍繁琐,本贴是对个人安装方法的记录。

推荐内容

​编辑​编辑​编辑

二、安装方法

安装要点:cuda、pytorch、NVIDIA CUB 版本的严格对应

安装所需链接:

  • Pytorch3d 各版本:Pytorch3d Releases Download
  • NVIDIA CUB 下载:NVIDIA/cub Download
  • 上述PyTorch官网历史版本链接( Previous PyTorch Versions | PyTorch)里面,找到v1.11.0的安装命令, 此处选择的是cuda11.3对应的PyTorch安装命令

安装前,首先要确定版本对应关系,本人的安装版本为:

CUDA Toolkit 11.3 + pytorch 1.11.0 + Visual Studio 2019 + Pytorch3d 0.6.2 + Python 3.10

1. 对于NVIDIA CUB, 其与CUDA Toolkit的版本关系如下图:pytorch3d,python,开发语言

 

本人安装cuda版本为11.3,所以CUB选择了对应的1.11.0版本,下载解压后将其路径添加到环境变量中:

pytorch3d,python,开发语言

 

2. 安装依赖项:

conda install -c fvcore -c iopath -c conda-forge fvcore iopath

3. 选择Pytorch3d版本,每个版本的Pytorch3d下有注明其适用的pytorch版本,如下图:

pytorch3d,python,开发语言

 

上图即pytorch3d 0.6.2版本适用的Pytorch版本为1.10.1, 1.10.2, 1.11.0,适用的Python为3.10, 但不再支持Pytorch1.6.0和Python3.6,其他版本的torch仍依据此下载对应的pytorch3d版本

4. 安装VS 2019后,打开如下图所示的"x64 Native Tools Command Prompt for VS 2019"终端,然后cd到pytorch3d解压后的目录路径里:

pytorch3d,python,开发语言
依次输入:

 在Windows上的命令提示符(Command Prompt)中设置环境变量:

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

在Windows上的PowerShell中设置环境变量:

$env:DISTUTILS_USE_SDK = 1

在Linux和macOS上设置环境变量:

export DISTUTILS_USE_SDK=1

then, 修改 pytorch3d里setup.py文件的源码:将extra_compile_args = {“cxx”: [“-std=c++14”]} 修改为: extra_compile_args = {“cxx”: []};pytorch3d,python,开发语言

 

5. 最后在"x64 Native Tools Command Prompt for VS 2019"终端输入:

python setup.py install --user

安装过程本人电脑上大约长达5分钟左右,页面不停的乱跳,最后的最后的最后最后显示成功:

pytorch3d,python,开发语言

如果你还安装失败,请联系我文章来源地址https://www.toymoban.com/news/detail-769002.html

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

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

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

相关文章

  • PyTorch3D 安装

    PyTorch 官网:PyTorch 官网 最新版本的 PyTorch:最新版本的 PyTorch 以 cuda 12.1,Python 3.10,Pytorch 2.1.2 为例 由于直接使用 pip 进行PyTorch3D 安装会报错,所以选择在官网直接下载对应的安装包进行安装。 进入 Anaconda 官网找到 PyTorch3D :PyTorch3D 官网 找到和 cuda 以及 Python 版本对应的

    2024年04月09日
    浏览(42)
  • 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)
  • pytorch3d成功安装

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

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

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

    2024年02月20日
    浏览(45)
  • linux 安装pytorch3d的坑

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

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

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

    2024年01月17日
    浏览(38)
  • 【深度学习】【三维重建】windows10环境配置PyTorch3d详细教程

    本人windows10下使用【Code for Neural Reflectance Surfaces (NeRS)】算法时需要搭建PyTorch3d环境,故此以详细教程以该算法依赖的环境版本为参照。 应很多朋友的要求,博主将自己的pytorch3d的conda环境打包放在【百度云盘;提取码:3tyw 】供大家参考,注意环境版本要求保持一致,并不适合

    2024年02月11日
    浏览(57)
  • 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日
    浏览(40)
  • pytorch3d安装遇到的一些坑和解决过程

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

    2024年02月06日
    浏览(37)
  • 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)

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

支付宝扫一扫打赏

博客赞助

微信扫一扫打赏

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

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

二维码1

领取红包

二维码2

领红包