【CUDA】Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

解決辦法

用tensorflow的時候都用下面三行:

config = tf.ConfigProto()
config.gpu_options.allow_growth = True
with tf.Session(config=config) as sess:
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章