前言

Pytorch3d 在windows上的安裝不如Linux系統(tǒng)方便，因?yàn)橐褂肅python去編譯C++的源碼，在CUB、cuda等軟件下，很容易出現(xiàn)問(wèn)題，今天給大家?guī)?lái)一個(gè)安裝方法和細(xì)節(jié)！

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

pytorch3d編譯安裝

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

---

1.編譯安裝

1.1.前置依賴(lài)和軟件環(huán)境（這里不講cuda安裝windows直接下載雙擊安裝就行，環(huán)境PATH自動(dòng)配置的）

如果沒(méi)有conda環(huán)境，創(chuàng)建conda 環(huán)境 安裝

conda create -n name python=3.9 conda activate pytorch3d

在環(huán)境中補(bǔ)充基本的軟件包

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 重要軟件版本安裝 ——本機(jī)環(huán)境參考

vs2022 && vs2019

cuda11.7(11.8不要選就行，因?yàn)樵?1.8的代碼重構(gòu)了，在編譯torch3d時(shí)候會(huì)有問(wèn)題)

conda torch安裝，進(jìn)入torch官網(wǎng)自行選擇，我是cuda11.7選擇如圖

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

這里還是要看清版本對(duì)應(yīng)的CUB，主要是你的CUDA版本，在官網(wǎng)查閱。
解壓到你的指標(biāo)目錄后，在系統(tǒng)PATH中建立CUB_HOME ，引入路徑：

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

1.3 安裝編譯和問(wèn)題解決

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

編譯步驟：打開(kāi)windows的菜單：

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

=
打開(kāi)后，進(jìn)入到你的torch3d目錄，然后激活你的conda環(huán)境，進(jìn)行編譯：

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

運(yùn)行：

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

等待編譯完成，過(guò)程中可能會(huì)遇到如下問(wèn)題：

問(wèn)題一：

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

問(wèn)題二：

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沒(méi)匹配，然后修改環(huán)境變量確定后，需要重啟VS的命令行 重復(fù)上述的編譯步驟，記得要重啟命令行！

問(wèn)題三：

error: namespace thrust::cub has no member DebugSyncStream
這個(gè)我是直接將11.8換成cuda11.7了 這個(gè)問(wèn)題不常見(jiàn)，不過(guò)記錄下，報(bào)錯(cuò)表面是語(yǔ)法問(wèn)題，本質(zhì)是cuda11.8的代碼重構(gòu)了部分。

---

問(wèn)題四：

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

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

安裝完成：

總結(jié)

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

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。