训练LaneATT遇到CUDA_HOME环境变量问题

我系统装了很久了,不记得之前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环境变量的问题,尚未搜到结果,然后有一个网页提示运行nvcc命令,但发现我电脑上没有该命令,提示需要安装 nvidia-cuda-toolkit

sudo apt install nvidia-cuda-toolkit

安装之后,依然没有CUDA_HOME环境变量,但此时执行 python setup.py install,直接安装成功了

纯属偏方

训练LaneATT遇到CUDA_HOME环境变量问题相关推荐

机器学习之 weka学习（一）weka介绍，安装和配置环境变量
本部分详情可查看博客http://blog.csdn.net/u011067360/article/details/20844443 ...
linux 环境变量设置（临时 + 永久）
一.环境变量环境变量是具有特殊名字的一个特定对象,包含了一个或多个应用程序运行所需的信息.(例如PATH,可执行程序的搜索路径,当要求系统运行一个程序,而没告诉系统它的具体路径时,系统就要在PTAH ...
训练MTCNN之前的环境以及依赖配置
训练MTCNN之前的环境以及依赖配置个人使用配置: win 10 Anaconda 4.2.0+python 3.5.5 vs2017 opencv 3.4.6 openblas 0.2.8 CUD ...
Ubuntu指令失效解决问题之一——错误配置环境变量
Ubuntu安装踩坑记录2: ·1. 突然修改错误环境变量,导致系统找不到指令,出现如下报错: Command 'dirname' is available in '/usr/bin/dirname' ...
linux环境oracle环境变量,Linux下设置oracle环境变量
Linux设置Oracle环境变量方法一:直接运行export命令定义变量,该变量只在当前的shell(BASH)或其子shell(BASH)下是有效的,shell关闭了,变量也就失效了,再打开新s ...
pytorch怎么加入环境变量_Pytorch源码编译简明指南
前言最近在浅尝Pytorch的源码,利用业余时间去品读品读,看着看着,第一次对Pytorch有了重新的认识. 原来现在Pytorch的版图是如此之大,Pytorch已经不是一年前的Pytorch了. ...
带你玩转Jetson Xavier NX系列教程 | Xavier NX 环境变量配置，风扇控制以及Jtop安装
写在最前:转载请注明文章出处.本期文章分为三部分内容,均为刷机以后需要配置的小细节,建议大家刷机以后立刻进行配置,没有配置的后续再配置也没什么大问题. Session_1 : : Xavier NX风 ...
Intel oneAPI学习笔记之环境变量配置运行一个PyTorch程序
前面两篇文章: Intel oneAPI学习笔记之基本概念&六大工具包一览 Intel oneAPI 学习笔记之 Centos 下安装 Intel® oneAPI Base Toolkit 和 ...
Texlive安装与环境变量配置
下载安装:https://www.jianshu.com/p/25896cc05cb4 去官网下载速度很快环境变量配置:https://www.jianshu.com/p/f8f6c255098c

训练LaneATT遇到CUDA_HOME环境变量问题

训练LaneATT遇到CUDA_HOME环境变量问题相关推荐

最新文章

热门文章