numba.cuda.cudadrv.error.NvvmSupportError: No supported GPU compute capabilities found

问题遇到的现象和发生背景

运行test.py时遇到的错误，cuda相关错误，奇怪的是运行train.py并没有报错

问题相关代码

numba.cuda.cudadrv.error.NvvmSupportError: No supported GPU compute capabilities found. Please check your cudatoolkit version matches your CUDA version.

运行结果及报错内容

我的解答思路和尝试过的方法

conda list查看了cudatoolkit版本等，觉得没问题，关键可以训练，就是测试时遇到这个问题，有点不明白

测试和训练需要的GPU能力可能不一样

numba版本的问题