专栏算法工具链numba.cuda.is_avalable() = False

numba.cuda.is_avalable() = False

yulus2023-08-02
22
2

在J5工具链1.1.60与1.1.57版本的环境中

复现pointpillar-qat的时候发现在validate阶段只能使用cpu进行计算mAP，发现原因是numba.cuda.is_avalable()为False，且输入nvcc -V 报错

在1.1.49b版本中是正常的

算法工具链
评论1
0/1000
  • 颜值即正义
    Lv.2
    您好，从OE1.1.52以后，Docker 环境中的 cudatoolkit 不再包含 nvvm 模块，所以numna无法使用cuda计算，如果您有在cuda上进行validate的需求，请使用1.1.52 docker或者其前序版本。
    2023-08-02
    0
    1
    • yulus回复颜值即正义:

      好的，谢谢

      2023-08-03
      0