failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED

报错提示：failed to create cublas handle: CUBLAS_STATUS_NOT_INITIALIZED

今天跑yolov4程序时，在predict图片时，出现了如上的报错。
查阅资料后发现，该报错是指没有分配显存或显存不够的时候才会报。

##在用gpu跑程序时，如果没有手动分配显存的话，默认能占多少占多少，但是占用过多可能会导致系统运行出错。所以给我们整了个报错.解决方案参考的该博主的config的说明，链接贴在后面。代码如下（用的tf2.4版本的）：

physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)

将该段代码贴在程序“导入”（也就是import***或者from***import***）后面即可.
该代码的意思是：
第一行查询该电脑有几张gpu可以用.
第二行的意思是需要用到显存时再申请，需要用多少再申请多少.

小哥的总结博文地址点此