CUDA_HOME environment variable is not set

[aswordok]

(trellis) D:+AI\myTrellis\TRELLIS\trellis>git clone --recurse-submodules https://github.com/JeffreyXiang/diffoctreerast.git ./tmp/extensions/diffoctreerast
(trellis) D:+AI\myTrellis\TRELLIS\trellis>pip install ./tmp/extensions/diffoctreerast
Processing d:+ai\mytrellis\trellis\trellis\tmp\extensions\diffoctreerast
Preparing metadata (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [12 lines of output]
Traceback (most recent call last):
File "", line 2, in
File "", line 34, in
File "D:+AI\myTrellis\TRELLIS\trellis\tmp\extensions\diffoctreerast\setup.py", line 10, in
CUDAExtension(
File "D:+AI\miniconda3\envs\trellis\lib\site-packages\torch\utils\cpp_extension.py", line 1078, in CUDAExtension
library_dirs += library_paths(cuda=True)
File "D:+AI\miniconda3\envs\trellis\lib\site-packages\torch\utils\cpp_extension.py", line 1216, in library_paths
paths.append(_join_cuda_home(lib_dir))
File "D:+AI\miniconda3\envs\trellis\lib\site-packages\torch\utils\cpp_extension.py", line 2416, in _join_cuda_home
raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed

× Encountered error while generating package metadata.
╰─> See above for output.

note: This is an issue with the package mentioned above, not pip.
hint: See above for details.

(trellis) D:+AI\myTrellis\TRELLIS\trellis>nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2024 NVIDIA Corporation
Built on Thu_Sep_12_02:55:00_Pacific_Daylight_Time_2024
Cuda compilation tools, release 12.6, V12.6.77
Build cuda_12.6.r12.6/compiler.34841621_0

(trellis) D:+AI\myTrellis\TRELLIS\trellis>python -m pip list | findstr torch
torch 2.5.1
torchvision 0.20.1

(trellis) D:+AI\myTrellis\TRELLIS\trellis>echo %CUDA_HOME%
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v12.6

I've actually set the variable CUDA_HOME

[aswordok]

上面是用 cmd.exe 环境运行的，换到 powershell 就好了。

[MengQiu-zl]

为什么我用powershell执行pip install ./tmp/extensions/diffoctreerast还是报错
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [14 lines of output]
Traceback (most recent call last):
File "", line 2, in
File "", line 34, in
File "D:\python_projects\TRELLIS-main\tmp\extensions\diffoctreerast\setup.py", line 10, in
CUDAExtension(
File "D:\miniconda3\Lib\site-packages\torch\utils\cpp_extension.py", line 1078, in CUDAExtension
library_dirs += library_paths(cuda=True)
^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\miniconda3\Lib\site-packages\torch\utils\cpp_extension.py", line 1216, in library_paths
paths.append(_join_cuda_home(lib_dir))
^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\miniconda3\Lib\site-packages\torch\utils\cpp_extension.py", line 2416, in _join_cuda_home
raise OSError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
[end of output]

note: This error originates from a subprocess, and is likely not a problem with pip.
error: metadata-generation-failed