解決tensorflow 調用bug Running model failed:Invalid argument: NodeDef mentions attr 'dilations'

在訓練DeepLab的時候,出現tensorflow 調用bug :

1. Running model failed:Invalid argument: NodeDef mentions attr 'dilations' not in Op<name=Conv2D; signature=input:T, filter:T ->

2. Executor failed to create kernel.

經過查找,就是tensorflow的版本過低。

解決方法:pip install -i https://pypi.tuna.tsinghua.edu.cn/simple tensorflow-gpu==1.10

發佈了28 篇原創文章 · 獲贊 12 · 訪問量 3萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章