设为首页
收藏本站
切换到宽版
登录
注册
找回密码
搜索
搜索
本版
文章
帖子
用户
快捷导航
易学开发平台
易学在线课堂
易学智能论坛
BBS
易学智能
»
易学智能论坛
›
技术大讨论
›
Tensorflow
›
could not create cudnn handle: CUDNN_STATUS_INTERNAL ...
返回列表
查看:
2088
|
回复:
2
could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR
[复制链接]
shaoheshaohe
shaoheshaohe
当前离线
积分
6670
665
主题
1234
帖子
6670
积分
xdtech
积分
6670
发消息
发表于 2019-2-9 20:53:13
|
显示全部楼层
|
阅读模式
在CPU下运行没有问题,但是在GPU版本下,一直运行不成功。可以尝试限制一下GPU的运行资源,添加如下代码:
config = tf.ConfigProto(gpu_options=tf.GPUOptions(allow_growth=True))
sess = tf.Session(config=config)
刚开始分配少量资源,然后按需慢慢增加GPU资源
---------------------
回复
使用道具
举报
Happy清子
Happy清子
当前离线
积分
11590
166
主题
616
帖子
1万
积分
xdtech
积分
11590
发消息
发表于 2019-5-18 08:12:28
|
显示全部楼层
意思是,
gpu占用太多,爆了
回复
使用道具
举报
Happy清子
Happy清子
当前离线
积分
11590
166
主题
616
帖子
1万
积分
xdtech
积分
11590
发消息
发表于 2019-5-18 08:12:51
|
显示全部楼层
需要找到一个利用率的平衡点
回复
使用道具
举报
返回列表
高级模式
B
Color
Image
Link
Quote
Code
Smilies
您需要登录后才可以回帖
登录
|
立即注册
本版积分规则
发表回复
回帖后跳转到最后一页
快速回复
返回顶部
返回列表