【深度学习】【三维重建】windows10环境配置PyTorch3d详细教程

【深度学习】【三维重建】windows10环境配置PyTorch3d详细教程

文章目录


前言

本人windows10下使用【Code for Neural Reflectance Surfaces (NeRS)】算法时需要搭建PyTorch3d环境,故此以详细教程以该算法依赖的环境版本为参照。

win11版本不能使用CUDA 10.2版本,可以参考博主的【Windows11下NeRS官方代码Pytorch实现

确定版本对应关系

windows10环境配置详细教程
环境版本要求保持一致:CUDA,CUB,Pytorch,Pytorch3d

注意这里的cub对应的是真实的物理机安装的cuda版本号(环境变量里配置的cuda),不是虚拟环境下的cuda版本号,否则即使pytorch3d安装成功,也不能使用gpu。

本人的安装版本为:

Python 3.8 + Pytorch 1.7.1 + CUDA Toolkit 10.2 + CUB1 .10.0 + Pytorch3d 0.5.0 +

使用anaconda3新建PyTorch3d虚拟环境:

bash 复制代码
conda create -n pytorch3d python=3.8
activate pytorch3d

然后安装对应版本pytorch和cuda包:

bash 复制代码
conda install -c pytorch pytorch=1.7.1 torchvision=0.8.2 cudatoolkit=10.2

博主的cuda版本为10.2,NVIDIA CUB推荐1.10.0,下载解压后将其路径添加到环境变量中:

注意不是在Path中添加

安装Pytorch3d的依赖项:

bash 复制代码
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

源码编译安装Pytorch3d

选择Pytorch3d版本,各版本下有注明其适用的pytorch版本,博主这里是0.5.0版本。

安装VS 2019,以管理者身份打开"x64 Native Tools Command Prompt for VS 2019"终端,cd到pytorch3d解压目录里,输入以下指令:

bash 复制代码
activate pytorch3d
set DISTUTILS_USE_SDK=1
set PYTORCH3D_NO_NINJA=1

修改pytorch3d里setup.py文件的源码:

python 复制代码
extra_compile_args = {"cxx": ["-std=c++14"]}
# 修改为
extra_compile_args = {"cxx": []}

继续在"x64 Native Tools Command Prompt for VS 2019"执行以下命令:

bash 复制代码
# 可能需要科学上网,需要下载一些依赖.
python setup.py install

成功安装。

bash 复制代码
# 查看环境中的是否成功安装包
conda list

总结

尽可能简单、详细的介绍windows10环境配置PyTorch3d详细教程,后续可以在当前配置的PyTorch3d环境中运行三维重建相关的代码。

相关推荐
CoovallyAIHub16 小时前
中科大DSAI Lab团队多篇论文入选ICCV 2025,推动三维视觉与泛化感知技术突破
深度学习·算法·计算机视觉
CoovallyAIHub16 小时前
开源的消逝与新生:从 TensorFlow 的落幕到开源生态的蜕变
pytorch·深度学习·llm
CoovallyAIHub1 天前
港大&字节重磅发布DanceGRPO:突破视觉生成RLHF瓶颈,多项任务性能提升超180%!
深度学习·算法·计算机视觉
CoovallyAIHub1 天前
英伟达ViPE重磅发布!解决3D感知难题,SLAM+深度学习完美融合(附带数据集下载地址)
深度学习·算法·计算机视觉
惯导马工2 天前
【论文导读】ORB-SLAM3:An Accurate Open-Source Library for Visual, Visual-Inertial and
深度学习·算法
隐语SecretFlow3 天前
国人自研开源隐私计算框架SecretFlow,深度拆解框架及使用【开发者必看】
深度学习
Billy_Zuo3 天前
人工智能深度学习——卷积神经网络(CNN)
人工智能·深度学习·cnn
羊羊小栈3 天前
基于「YOLO目标检测 + 多模态AI分析」的遥感影像目标检测分析系统(vue+flask+数据集+模型训练)
人工智能·深度学习·yolo·目标检测·毕业设计·大作业
l12345sy3 天前
Day24_【深度学习—广播机制】
人工智能·pytorch·深度学习·广播机制
九章云极AladdinEdu3 天前
超参数自动化调优指南:Optuna vs. Ray Tune 对比评测
运维·人工智能·深度学习·ai·自动化·gpu算力