tensorflow报错Cannot assign a device for operation xxxxxx Could not satisfy explicit device解决方法
tensorflow
一个面向所有人的开源机器学习框架
项目地址:https://gitcode.com/gh_mirrors/te/tensorflow
免费下载资源
·
tensorflow报错Cannot assign a device for operation xxxxxx Could not satisfy explicit device specification '/device:GPU:0' because no supported kernel for GPU devices is available的解决方法
前情提要
在使用tensorflow运行ImageNet时报错
Cannot assign a device for operation ‘InceptionV3/AuxLogits/Conv2d_2b_1x1/weights/RMSProp_1’: Could not satisfy explicit device specification ‘/device:GPU:0’ because no supported kernel for GPU devices is available.
有时报错CUDNN_STATUS_NOT_INITIALIZED,原因不明,但已经确认CUDA版本和CUDNN版本正确并且正确配置。
我的配置
- Windows 10 家庭中文版(64位)
- I7 8750H@2.2GHz
- 16G DDR4
- GTX1050ti(4G)
- python3.5+tensorflow_gpu-1.4.0
- CUDA8.0+CUDNN6.0
解决方法
#修改代码
###########################
# Kicks off the training. #
###########################
config=tf.ConfigProto(allow_soft_placement=True)#修改这里
slim.learning.train(
train_tensor,
logdir=FLAGS.train_dir,
master=FLAGS.master,
is_chief=(FLAGS.task == 0),
init_fn=_get_init_fn(),
summary_op=summary_op,
number_of_steps=FLAGS.max_number_of_steps,
log_every_n_steps=FLAGS.log_every_n_steps,
save_summaries_secs=FLAGS.save_summaries_secs,
save_interval_secs=FLAGS.save_interval_secs,
sync_optimizer=optimizer if FLAGS.sync_replicas else None,
session_config=config)
修改后错误消除,可以正常运行。
共同学习,一起进步!
GitHub 加速计划 / te / tensorflow
184.55 K
74.12 K
下载
一个面向所有人的开源机器学习框架
最近提交(Master分支:2 个月前 )
a49e66f2
PiperOrigin-RevId: 663726708
2 个月前
91dac11a
This test overrides disabled_backends, dropping the default
value in the process.
PiperOrigin-RevId: 663711155
2 个月前
更多推荐
已为社区贡献6条内容
所有评论(0)