tensorflow踩坑
雷咖啡
程序之路才刚开始
展开
-
Tensorflow版本兼容问题(keras,cuda,cudnn)
Tensorflow使用GPU进行训练之前,需要确认的是GPU是否可用如不确定机器是否有GPU,可以输入以下命令进行确认查看显卡信息: lspci | grep -i vgasudo prime-select nvidia # 切换nvidia显卡sudo prime-select intel # 切换intel显卡sudo prime-select query # 查看当前使...原创 2019-11-05 22:41:07 · 1555 阅读 · 0 评论 -
tensorflow踩坑记录
Input 'b' of 'MatMul' Op has type float32 that does not match type int32 of argument 'a'tf.reset_default_graphwith cnn_graph.as_default(): with tf.name_scope("placeholders"): inputs =...原创 2018-10-22 15:07:54 · 872 阅读 · 0 评论