windows安装pytorch3d（2024年8月 ）

### PyTorch3D在Windows上的安装 #### 创建Python环境 为了确保兼容性和稳定性，在开始之前应该先创建一个新的Conda环境。这可以通过下面的命令完成： ```bash conda create -n pytorch3d python=3.9 conda activate pytorch3d ``` #### 安装CUDA和cuDNN 对于GPU加速的支持，需要预先配置好相应的CUDA版本以及配套的cuDNN库[^1]。 #### 安装CUB库 接着需获取并设置[CUB](https://github.com/NVIDIA/cub/releases/tag/1.9.10) 库至当前环境中。建议将其放置于`pytorch3d`虚拟环境目录内以防误删除[^2]。 #### 修改PyTorch3D源码适应本地编译需求 下载指定版本(如v0.7.2) 的PyTorch3D压缩包后解压，并编辑其中的`setup.py`文件来调整编译参数。具体操作是在该脚本中定位到`extra_compile_args={"cxx": [...]}`这一行并将方括号内的内容移除。 #### 编译与安装PyTorch3D 使用Visual Studio 2019自带的x64原生工具命令提示符作为终端执行如下指令来进行最终的构建过程。需要注意的是路径部分应当替换为实际存放位置： ```batchfile cd C:\path\to\your\unzipped_pytorch3d_folder activate pytorch3d set DISTUTILS_USE_SDK=1 set PYTORCH3D_NO_NINJA=1 python setup.py install ``` 尽管有文档推荐采用VS命令行界面，实际上也可以通过Anaconda Prompt (Miniconda3) 来激活环境并运行上述相同步骤[^3]。 ### 使用PyTorch3D 一旦成功完成了以上所有准备工作，则可以在任何支持此环境的地方导入并利用PyTorch3D模块开展工作了。