第四步为什么不行 #7
Description
(base) u2025170778@n1:~/jupyterlab/track5$ python setup.py develop
Traceback (most recent call last):
File "/home/u2025170778/jupyterlab/track5/setup.py", line 59, in
make_cuda_ext(
File "/home/u2025170778/jupyterlab/track5/setup.py", line 18, in make_cuda_ext
cuda_ext = CUDAExtension(
^^^^^^^^^^^^^^
File "/home/u2025170778/.local/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1337, in CUDAExtension
library_dirs += library_paths(device_type="cuda")
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/u2025170778/.local/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 1548, in library_paths
if (not os.path.exists(_join_cuda_home(lib_dir)) and
^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/u2025170778/.local/lib/python3.11/site-packages/torch/utils/cpp_extension.py", line 2982, 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.