win10系统下安装PyTorch3d
最近在复现3d人体重建和人脸重建相关demo算法,发现都需要安装PyTorch3d,然而Linux环境借助Conda可以一键安装,对于Win10环境来说真实太痛苦了,博主折腾了几天,各种重建环境,现在已成功安装PyTorch3d。一、环境版本一致性要求环境版本最好保持一致性,省的来回更改环境,以至于最后把系统环境改的面目全非。NVIDIA CUDA,NVIDIA CUB,Visual Studio
最近在复现3d人体重建和人脸重建相关demo算法,发现都需要安装PyTorch3d,然而Linux环境借助Conda可以一键安装,对于Win10环境来说真实太痛苦了,博主折腾了几天,各种重建环境,现在已成功安装PyTorch3d。
一、环境版本一致性要求
环境版本最好保持一致性,省的来回更改环境,以至于最后把系统环境改的面目全非。
NVIDIA CUDA,NVIDIA CUB,Visual Studio ,Pytorch,Pytorch3d必须需要版本一致
首先官网对于Pytorch3d在windows安装说明

本人显卡为3090,NVIDIA CUDA是cuda11.3,Conda中pytorch=1.8.0,cuda=11.1,Visual Studio =2019,Pytorch3d==0.6.0;NVIDIA CUB=1.11.0
其中NVIDIA CUDA11.3支持Visual Studio2017和2019,安装2022会报版本不一致,所以最好还是老老实实安装2019,其中NVIDIA CUB官网推荐1.10.0,这里需要更改为1.11.0不然会报版本不一致。


注意:
- 把setup.py 中[‘ninja’, ‘-v’]改为[‘ninja’, ‘–version’]是错误的
- 不用在系统环境变量中配置Visual Studio2019 的cl.exe,直接通过下图所示启动命令窗口,然后输入set DISTUTILS_USE_SDK=1,同时进入pytorch3d文件夹:


二、提前修改相关文件
1.pytoch3d中源码更改
setup.py 中43行:
extra_compile_args = {"cxx": ["-std=c++14"]}注释掉改为extra_compile_args = {"cxx": []}

setup.py 中60行:-std=c++14注释掉

iou_box3d.cu修改注释掉>#include "utils/pytorch3d_cutils.h"

iou_utils.cuh和iou_utils.h搜索not,把not改为!,最新版本PyTorch3d已经解决这个问题
更改后可以解决如下错误:
2.pytorch源码中更改
需要找到cpp_extension.py进行修改,本人如下:
C:\Users\dell\miniconda3\envs\icon\lib\site-packages\torch\utils\cpp_extension.py

三、命令窗口安装编译
在命令窗口输入 python setup.py install 进行安装编译

四、参考
- https://blog.csdn.net/zzqkz20121221/article/details/121157357
- https://github.com/facebookresearch/pytorch3d/issues/713
- https://zhuanlan.zhihu.com/p/371279126
- https://mp.csdn.net/mp_blog/creation/editor/125011964
- https://zhuanlan.zhihu.com/p/371279126
- https://blog.csdn.net/qq_40714949/article/details/116571010
- https://blog.csdn.net/weixin_41793473/article/details/118669832
更多推荐



所有评论(0)