tensorflow
白夜泊
这个作者很懒,什么都没留下…
展开
-
使用tensorflow时报错Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
使用tf2.1时,cuda和cudnn都安装好了,报错 Could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR 最开始以为是cudnn版本问题,试了各种方法,包括用conda重新装,不过推荐大家安装tensorflow-gpu可以用conda,可以默认安装cuda和cudnn,省心。不过这个错误不是这个原因导致的。 解决办法是在代码中加入下面代码: pus = tf.config.experimental.list_physical_devic原创 2020-05-29 10:13:21 · 1434 阅读 · 1 评论 -
UnknownError:Fail to find the dnn implementation解决方法
程序加上下面代码 gpus = tf.config.experimental.list_physical_devices(‘GPU’) if not gpus: return “No GPU available” try: # Currently, memory growth needs to be the same across GPUs for gpu in gpus: tf.config.e...原创 2019-11-17 20:28:22 · 2588 阅读 · 1 评论