NVlabs / CF-3DGS

Other
394 stars 43 forks source link

Can't install pytorch3D #13

Closed jjas-min closed 4 months ago

jjas-min commented 4 months ago

I proceeded with anaconda environment setting in the following order indicated on readme.md

conda create -n cf3dgs python=3.10 conda activate cf3dgs conda install conda-forge::cudatoolkit-dev=11.7.0 conda install pytorch==2.0.0 torchvision==0.15.0 pytorch-cuda=11.7 -c pytorch -c nvidia pip install -r requirements.txt

However, the following error occurs when installing pytorch3d at the end.

note: This error originates from a subprocess, and is likely not a problem with pip. ERROR: Failed building wheel for pytorch3d Running setup.py clean for pytorch3d Failed to build pytorch3d ERROR: Could not build wheels for pytorch3d, which is required to install pyproject.toml-based projects

has anyone fixed it?

willyawan16 commented 4 months ago

I get the same issue too

willyawan16 commented 4 months ago

After a few hours, I found out that i need to add nvidia cub to the environment variable The error that I got that made the pytorch3d build fail was this: _UserWarning: The environment variable CUB_HOME was not found. NVIDIA CUB is required for compilation and can be downloaded from https://github.com/NVIDIA/cub/releases. You can unpack it to a location of your choice and set the environment variable CUB_HOME to the folder containing the CMakeListst.txt file._