运行ObjectDetection-OneStageDet里的make时报错

news/2024/7/10 3:10:48 标签: yolo, 目标检测, pytorch

按照腾讯优图实验室开源的One-Stage目标检测项目里的教程进行复现时,进行到Preparation里的make -j32时报错
raise EnvironmentError('The nvcc binary could not be located in your $PATH. Either add it to your path, or set $CUDAHOME')
看了下应该是环境变量有问题,但添加了好几次都不得其法,最后暴力解决:
把setup.py代码里


    # first check if the CUDAHOME env variable is in use
    if 'CUDAHOME' in os.environ:
        home = os.environ['CUDAHOME']
        nvcc = pjoin(home, 'bin', 'nvcc')
    else:
        # otherwise, search the PATH for NVCC
        default_path = pjoin(os.sep, 'usr', 'local', 'cuda', 'bin')
        nvcc = find_in_path('nvcc', os.environ['PATH'] + os.pathsep + default_path)
        if nvcc is None:
            raise EnvironmentError('The nvcc binary could not be '
                'located in your $PATH. Either add it to your path, or set $CUDAHOME')
        home = os.path.dirname(os.path.dirname(nvcc))

    cudaconfig = {'home':home, 'nvcc':nvcc,
                  'include': pjoin(home, 'include'),
                  'lib64': pjoin(home, 'lib64')}
    for k, v in cudaconfig.items():
        if not os.path.exists(v):
            raise EnvironmentError('The CUDA %s path could not be located in %s' % (k, v))

    return cudaconfig
CUDA = locate_cuda()

这一段中的if里的内容放出来,并把

home = os.environ['CUDAHOME']

改成

home=CUDAPATH

其中CUDAPATH是你自己安装的cuda的路径,如"/opt/soft/cuda/".

解决。


http://www.niftyadmin.cn/n/1168658.html

相关文章

为什么走出舒适区,ALL in区块链

2019独角兽企业重金招聘Python工程师标准>>> 大家好,我是Bob Jiang,HiBlock区块链社区发起人;中国北方的第一位CST(Certified Scrum Trainer),国内的敏捷(Agile)大咖&…

神经网络压缩库Distiller学习

课题需要来学习一下Intel开源的Distiller神经网络压缩库。 Intel 主要根据以下特征和工具构建了 Distiller: 集成了剪枝、正则化和量化算法的框架分析和评估压缩性能的一组工具当前最优压缩算法的示例实现 安装 Clone Distiller git clone https://github.com.…

服务器(2080Ti)配置pytorch,distiller开发环境踩坑记录

公司计算资源限制,将开发环境转移到学校实验室的服务器上进行,需要重新配置一遍开发环境。服务器配置为10块RTX 2080Ti的显卡。本以为有了之前一次配置环境的经验会很顺利,naive。。。 1.torch包之间的依赖关系 由于distiller的requirement…

SpringBoot 返回html

2019独角兽企业重金招聘Python工程师标准>>> 最近需要SpringBoot直接返回html、js等静态资源文件,在网上找了很久没有解决我的问题。 网上大部分的教程基本上都是关于模板框架的,如TheamLeaf,而我并不需要这个。 终于找到一篇博客…

pytorch使用多GPU进行训练

首先需要在代码开头注明所使用的GPU序号,比如: import torch.nn as nn import osos.environ["CUDA_VISIBLE_DEVICES"] 0,1对linux系统来说,可以使用 watch -n 0.1 nvidia-smi来查看服务器上GPU的状态与可用GPU序号。…

【luoguP4006 清华集训2017】小Y和二叉树

题目描述 小 Y 是一个心灵手巧的 OIer,她有许多二叉树模型。 小 Y 的二叉树模型中,每个结点都具有一个编号,小 Y 把她最喜欢的一个二叉树模型挂在了墙上,树根在最上面,左右子树分别在树根的左下方与右下方,…

Distiller tutorial: Pruning Filters Channels

本文为对distiller教程 Pruning Filters & Channels 的翻译。 原文地址:Pruning Filters & Channels Introduction channel和filter的剪枝是结构化剪枝的示例,这些剪枝方法不需要特殊硬件即可完成剪枝并压缩模型,这也使得这种剪枝…

python可变参数*args 和 **kwargs

可变变量并不是一定要写成*args 和**kwargs,只有变量名称前的*才是必须的,如果你喜欢,可以写成*var和**vars,而写成*args 和**kwargs只是一个通俗的命名约定。 *args的用法 *args和**kwargs主要用于函数定义。 你可以将不定数量的…