1. 环境准备:搭建稳定基础
在Windows 11上部署PyTorch3D,环境配置是第一步也是最重要的一步。我遇到过太多因为环境问题导致的安装失败案例,所以这里会详细说明每个环节的注意事项。
首先需要确认你的Windows 11系统版本。建议使用22H2或更高版本,因为早期版本可能存在一些兼容性问题。我实测过在21H2上会遇到奇怪的路径处理问题,升级到23H2后这些问题都消失了。检查系统版本的方法很简单:Win+R输入winver,确保版本号不低于10.0.22621。
硬件方面,你需要一块支持CUDA的NVIDIA显卡。虽然PyTorch3D理论上可以在CPU上运行,但实际开发中GPU加速是必不可少的。运行dxdiag命令,在"显示"选项卡中确认你的显卡型号。我建议至少使用RTX 2060及以上显卡,因为3D渲染对显存要求较高,6GB显存是最低配置。
开发工具链的选择很关键:
- Visual Studio 2022 Community版(必须安装"使用C++的桌面开发"工作负载)
- CUDA Toolkit 12.1(与PyTorch版本严格对应)
- cuDNN 8.9(需要注册NVIDIA开发者账号下载)
- Anaconda或Miniconda(推荐后者,更轻量)
安装Visual Studio时有个坑要注意:默认安装会缺少Windows 10/11 SDK,这会导致后续编译失败。我建议在VS安装界面勾选:
- MSVC v143 - VS 2022 C++ x64/x86生成工具
- Windows 10/11 SDK(10.0.22621.0)
- C++ CMake工具
- 测试适配器核心功能
2. 创建Python虚拟环境
Python环境管理是避免依赖冲突的关键。我强烈建议使用conda而不是pip直接安装,因为PyTorch3D的依赖关系比较复杂。
创建环境的命令看似简单:
conda create -n pytorch3d python=3.11
conda activate pytorch3d
但这里有三个细节需要注意:
- Python版本选择:官方推荐3.9-3.11,我实测3.11.9最稳定
- 环境名称不要包含特殊字符和空格
- 创建环境时不要添加任何额外包(如numpy),这可能导致后续冲突
安装基础依赖时,顺序很重要。先安装fvcore和iopath:
conda install -c fvcore -c iopath -c conda-forge fvcore iopath

382

被折叠的 条评论
为什么被折叠?



