查看: 2088|回复: 2

could not create cudnn handle: CUDNN_STATUS_INTERNAL_ERROR

[复制链接]

665

主题

1234

帖子

6670

积分

xdtech

Rank: 5Rank: 5

积分
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资源
---------------------


回复

使用道具 举报

166

主题

616

帖子

1万

积分

xdtech

Rank: 5Rank: 5

积分
11590
发表于 2019-5-18 08:12:28 | 显示全部楼层
意思是,
gpu占用太多,爆了
回复

使用道具 举报

166

主题

616

帖子

1万

积分

xdtech

Rank: 5Rank: 5

积分
11590
发表于 2019-5-18 08:12:51 | 显示全部楼层
需要找到一个利用率的平衡点
回复

使用道具 举报

您需要登录后才可以回帖 登录 | 立即注册

本版积分规则

快速回复 返回顶部 返回列表