IT数码 购物 网址 头条 软件 日历 阅读 图书馆
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
图片批量下载器
↓批量下载图片,美女图库↓
图片自动播放器
↓图片自动播放器↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁
 
   -> Python知识库 -> Windows下Pytorch3d的安装方法 -> 正文阅读

[Python知识库]Windows下Pytorch3d的安装方法

一、问题:

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

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

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

二、安装方法

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

安装所需链接:

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

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

1. 对于NVIDIA CUB, 其与CUDA Toolkit的版本关系如下图:

在这里插入图片描述

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

在这里插入图片描述

2. 安装依赖项:

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

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

在这里插入图片描述

上图即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解压后的目录路径里:

在这里插入图片描述
依次输入:

set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

then, 修改 pytorch3d里setup.py文件的源码:将extra_compile_args = {“cxx”: [“-std=c++14”]} 修改为: extra_compile_args = {“cxx”: []};
在这里插入图片描述

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

python setup.py install

安装过程本人电脑上大约5分钟左右,最后显示成功:
在这里插入图片描述

  Python知识库 最新文章
Python中String模块
【Python】 14-CVS文件操作
python的panda库读写文件
使用Nordic的nrf52840实现蓝牙DFU过程
【Python学习记录】numpy数组用法整理
Python学习笔记
python字符串和列表
python如何从txt文件中解析出有效的数据
Python编程从入门到实践自学/3.1-3.2
python变量
上一篇文章      下一篇文章      查看所有文章
加:2022-09-30 00:49:11  更:2022-09-30 00:50:55 
 
开发: C++知识库 Java知识库 JavaScript Python PHP知识库 人工智能 区块链 大数据 移动开发 嵌入式 开发工具 数据结构与算法 开发测试 游戏开发 网络协议 系统运维
教程: HTML教程 CSS教程 JavaScript教程 Go语言教程 JQuery教程 VUE教程 VUE3教程 Bootstrap教程 SQL数据库教程 C语言教程 C++教程 Java教程 Python教程 Python3教程 C#教程
数码: 电脑 笔记本 显卡 显示器 固态硬盘 硬盘 耳机 手机 iphone vivo oppo 小米 华为 单反 装机 图拉丁

360图书馆 购物 三丰科技 阅读网 日历 万年历 2024年11日历 -2024/11/15 6:49:22-

图片自动播放器
↓图片自动播放器↓
TxT小说阅读器
↓语音阅读,小说下载,古典文学↓
一键清除垃圾
↓轻轻一点,清除系统垃圾↓
图片批量下载器
↓批量下载图片,美女图库↓
  网站联系: qq:121756557 email:121756557@qq.com  IT数码