Open xgh127 opened 5 months ago
This is likely because your PyTorch has not been compiled with CUDA support, and it's not an issue with CUDA_HOME. I would suggest reinstalling PyTorch with CUDA support. The PyTorch codebase can be quite messy at times.
The error info is as below:I've already set the CUDA_HOME and checked it on my command line,but when I run setup.py,it still tells my that"CUDA_HOME environment variable is not set. Please set it to your CUDA install root."I wonder why????????? (dmodel) C:\Users\xgh\tiny-cuda-nn\bindings\torch>python setup.py install C:\Users\xgh\tiny-cuda-nn\bindings\torch\setup.py:5: DeprecationWarning: pkg_resources is deprecated as an API. See https://setuptools.pypa.io/en/latest/pkg_resources.html from pkg_resources import parse_version Building PyTorch extension for tiny-cuda-nn version 1.7 Obtained compute capabilities [86] from environment variable TCNN_CUDA_ARCHITECTURES nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2021 NVIDIA Corporation Built on Fri_Dec_17_18:28:54_Pacific_Standard_Time_2021 Cuda compilation tools, release 11.6, V11.6.55 Build cuda_11.6.r11.6/compiler.30794723_0 Detected CUDA version 11.6 Targeting C++ standard 17 Traceback (most recent call last): File "C:\Users\xgh\tiny-cuda-nn\bindings\torch\setup.py", line 187, in
ext_modules = [make_extension(comp) for comp in compute_capabilities]
File "C:\Users\xgh\tiny-cuda-nn\bindings\torch\setup.py", line 187, in
ext_modules = [make_extension(comp) for comp in compute_capabilities]
File "C:\Users\xgh\tiny-cuda-nn\bindings\torch\setup.py", line 172, in make_extension
ext = CUDAExtension(
File "C:\Users\xgh\anaconda3\envs\dmodel\lib\site-packages\torch\utils\cpp_extension.py", line 1074, in CUDAExtension
library_dirs += library_paths(cuda=True)
File "C:\Users\xgh\anaconda3\envs\dmodel\lib\site-packages\torch\utils\cpp_extension.py", line 1208, in library_paths
paths.append(_join_cuda_home(lib_dir))
File "C:\Users\xgh\anaconda3\envs\dmodel\lib\site-packages\torch\utils\cpp_extension.py", line 2407, 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.
(dmodel) C:\Users\xgh\tiny-cuda-nn\bindings\torch>echo %CUDA_HOME% C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.6