【Tensorflow】failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

问题描述

使用tensorflow的gpu时出现了报错:
查找原因发现是显存空间不走所导致的（其实具体原因还是money的问题）

解决问题

有两个主要原因
1、cudnn、cuda的版本与tensorflow不适用，但是我的是按照教程来的，确认好几遍，保证是可以的，这个排除
2、gpu的显存不足，可以通过官网上的方法解决：T因为ensorFlow2.0支持两种GPU运算的方式：
（1）动态分配显存
（2）设置硬性显存（比如限制只能用1g显存，其他的去玩游戏
把模式调成（1）动态分配即可，代码为；

import tensorflow as tf

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