前言

Pytorch3d 在windows上的安裝不如Linux系統方便，因為要使用Cpython去編譯C++的源碼，在CUB、cuda等軟件下，很容易出現問題，今天給大家帶來一個安裝方法和細節！

本方法支持windows10和win11系統，VS2022、VS2019等目前主流的系統和軟件版本，在本機環境中，由于我是剛剛裝的，所以除了CUDA都是最新版本，這里提前告訴下讀者們。

pytorch3d編譯安裝

• 前言
• 1.編譯安裝
• 1.1.前置依賴和軟件環境（這里不講cuda安裝windows直接下載雙擊安裝就行，環境PATH自動配置的）
• • 1.2 重要軟件版本安裝 ——本機環境參考
  • 1.3 安裝編譯和問題解決
  • 問題一：
  • 問題二：
  • 問題三：
  • 問題四：
  • • 安裝完成：

---

1.編譯安裝

1.1.前置依賴和軟件環境（這里不講cuda安裝windows直接下載雙擊安裝就行，環境PATH自動配置的）

如果沒有conda環境，創建conda 環境 安裝

conda create -n name python=3.9 conda activate pytorch3d

在環境中補充基本的軟件包

conda install -c fvcore -c iopath -c conda-forge fvcore iopath conda install jupyter pip install scikit-image matplotlib imageio plotly opencv-python pip install black usort flake8 flake8-bugbear flake8-comprehensions

1.2 重要軟件版本安裝 ——本機環境參考

vs2022 && vs2019

cuda11.7(11.8不要選就行，因為在11.8的代碼重構了，在編譯torch3d時候會有問題)

conda torch安裝，進入torch官網自行選擇，我是cuda11.7選擇如圖

CUB 選擇最新的即可：（我選的CUB1.17，1.71.2 我都試過）
地址：https://github.com/NVIDIA/cub/releases

這里還是要看清版本對應的CUB，主要是你的CUDA版本，在官網查閱。
解壓到你的指標目錄后，在系統PATH中建立CUB_HOME ，引入路徑：

點擊pytorch3d 下載：pytorch3d 下載source包
我下的0.7.0最新的，也是要核對好了版本,我的torch是1.12.1，所以還是下了最新的。

1.3 安裝編譯和問題解決

打開pytorch3D文件下的，setup.py文件，將extra_compile_args = {“cxx”: [“-std=c++14”]} 修改為: extra_compile_args = {“cxx”: []};

編譯步驟：打開windows的菜單：

一般VS會自動給你配好，不論VS什么版本，如果找不到一般在這里C:\ProgramData\Microsoft\Windows\Start
Menu\Programs|

=
打開后，進入到你的torch3d目錄，然后激活你的conda環境，進行編譯：

下面的問題一，你如果和我一樣安裝的新版本，那么必會遇到所以直接修改了再編譯！

運行：

set DISTUTILS_USE_SDK=1 set PYTORCH3D_NO_NINJA=1 python setup.py install

等待編譯完成，過程中可能會遇到如下問題：

問題一：

根據圖中編譯提示：我們打開你的cuda目錄下：\include\thrust\system\cuda\config的代碼,在74行加入提示的語句，如下修改：

問題二：

C:/Program Files/NVIDIA GPU Computing Toolkit/CUDA/v11.7/include\cub/device/dispatch/dispatch_segmented_sort.cuh(379): error: expected a member name3 errors detected in the compilation of "D:/research/code/pytorch3d/pytorch3d/csrc/pulsar/cuda/renderer.backward.gpu.cu". renderer.backward.gpu.cu

切換CUB的版本和你的cuda沒匹配，然后修改環境變量確定后，需要重啟VS的命令行 重復上述的編譯步驟，記得要重啟命令行！

問題三：

error: namespace thrust::cub has no member DebugSyncStream
這個我是直接將11.8換成cuda11.7了 這個問題不常見，不過記錄下，報錯表面是語法問題，本質是cuda11.8的代碼重構了部分。

---

問題四：

raise RuntimeError(message) RuntimeError: Error compiling objects for extension

打開：\envs\torch-gpu\Lib\site-packages\torch\utils 下的cpp_extension.cpp文件，搜索 ‘ninja’ 然后將[‘ninja’,‘-v’]改成[‘ninja’,‘–version’]， 嘗試切換CUB的版本把。如果你按照我的步驟，這個問題和問題二和問題三都是應該產生不了的。

安裝完成：

總結

以上是生活随笔為你收集整理的10.24 cv方向3DMM必备环境——Windows10/11下pytorch3d[完美安装版]的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。