GPU 版 TensorFlow failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED

原因：

使用 GPU 版 TensorFlow ，並且在顯卡高占用率的情況下（比如玩游戲）訓練模型，要注意在初始化 Session 的時候為其分配固定數量的顯存，否則可能會在開始訓練的時候直接報錯退出。

 

解決方法：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
sess = tf.Session(config=tf.ConfigProto(gpu_options=gpu_options))

 

原先代碼：

with tf.Session() as sess:
    pass

現在代碼：

gpu_options = tf.GPUOptions(per_process_gpu_memory_fraction=0.333)
with tf.Session(config=tf.ConfigProto(gpu_options=gpu_options)) as sess:
    pass