failed to create cublas handle: CUBLAS_STATUS_ALLOC_FAILED 错误解决方法

若是你是使用 GPU 版 TensorFlow 的话,而且你想在显卡高占用率的状况下(好比玩游戏)训练模型,那你要注意在初始化 Session 的时候为其分配固定数量的显存,不然可能会在开始训练的时候直接报错退出:python 2017-06-27 20:39:21.955486: E c:\tf_jenkins\home\workspace\release-win\m\windows-gpu\p
相关文章
相关标签/搜索