Переменная среды CUDA_HOME не работала ⇐ Python

[Гость]

Я студент, и когда я воспроизводил статью, я столкнулся со следующей ситуацией: Подскажите, пожалуйста, как мне ее изменить

Код: Выделить всё

PS D:\CorrI2P-main\CorrI2P-main - test2\index_max_ext> pip install .
Processing d:\corri2p-main\corri2p-main - test2\index_max_ext
Preparing metadata (setup.py) ... error
error: subprocess-exited-with-error

× python setup.py egg_info did not run successfully.
│ exit code: 1
╰─> [15 lines of output]
Traceback (most recent call last):
File "", line 2, in 
File "", line 34, in 
File "D:\CorrI2P-main\CorrI2P-main - test2\index_max_ext\setup.py", line 17, in 
ext_modules=[CUDAExtension('index_max', ['index_max.cpp', 'index_max_cuda.cu'])],
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Python\Lib\site-packages\torch\utils\cpp_extension.py", line 1074, in CUDAExtension
library_dirs += library_paths(cuda=True)
^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Python\Lib\site-packages\torch\utils\cpp_extension.py", line 1208, in library_paths
paths.append(_join_cuda_home(lib_dir))
^^^^^^^^^^^^^^^^^^^^^^^^
File "D:\Python\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.
[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.

I tried adding the CUDA_HOME environment variable, but it didn't work.


Источник: https://stackoverflow.com/questions/781 ... didnt-work