我系統(tǒng)裝了很久了,不記得之前CUDA是咋裝了,然后訓練神經網絡一直也沒啥問題,今天想學習LaneATT,車道線識別的項目,需要先安裝nms,遇到如下問題:

(base) luke@luke-NH5x-7xRCx-RDx:~/Download/study/LaneATT/lib/nms$ python setup.py 
Traceback (most recent call last):File "setup.py", line 7, in <module>ext_modules=[CUDAExtension('nms.details', ['src/nms.cpp', 'src/nms_kernel.cu'])],File "/home/luke/miniconda3/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 912, in CUDAExtensionlibrary_dirs += library_paths(cuda=True)File "/home/luke/miniconda3/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 1020, in library_pathsif (not os.path.exists(_join_cuda_home(lib_dir)) andFile "/home/luke/miniconda3/lib/python3.7/site-packages/torch/utils/cpp_extension.py", line 2035, in _join_cuda_homeraise EnvironmentError('CUDA_HOME environment variable is not set. '
OSError: CUDA_HOME environment variable is not set. Please set it to your CUDA install root.
(base) luke@luke-NH5x-7xRCx-RDx:~/Download/study/LaneATT/lib/nms$ 

網上搜素怎么解決CUDA_HOME環(huán)境變量的問題,尚未搜到結果,然后有一個網頁提示運行nvcc命令,但發(fā)現(xiàn)我電腦上沒有該命令,提示需要安裝?nvidia-cuda-toolkit

sudo apt install nvidia-cuda-toolkit

安裝之后,依然沒有CUDA_HOME環(huán)境變量,但此時執(zhí)行 python setup.py install,直接安裝成功了

純屬偏方

總結

以上是生活随笔為你收集整理的训练LaneATT遇到CUDA_HOME环境变量问题的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。