2017-01-19 4 views
0

vgg-19モデルを使用してスタイル転送を実装しようとしました。バックエンドとしてテンソルフローを使用してvgg-19の既存のkerasモデルを使って重みをロードしました。まずノートブックで試しましたが、tf 0.11.0では非常に遅いですが、うまくいきました。次に、バージョン0.12.1のgpusでAWSインスタンスに切り替え、同じスクリプトを試しました.Jupyterノートではエラーは発生しませんが、代わりにカーネルがフリーズし、出力を計算しようとすると強制的に再起動されます第1の層の入力を除く特定の層。私は0.11.0のgpuバージョンをインストールしました - 両方のバージョンが行列乗算のテストに合格しましたが、同じ問題です。Pythonコンソールにログがないsess.run()でTensorflowがクラッシュする

また、0.11.0用に書かれたvgg-16を使用して誰かのスクリプトで実行しようとしましたが、層のサイズがエラーのためセッションがクラッシュしました。 tf 0.11.0のGPUバージョン - 同じフリーズと再起動。

pythonコンソールのものを除いて、tensorflowによって書き込まれたログはありますか?私はそれらについて何の言及も見つけられませんでした

答えて

0

固定されました。 cudnnのバージョンがテンソルフローのバージョンと一致しませんでした。エラー出力は、pythonのコンソールにのみ出力され、jupiterのノートには出力されませんでした。

関連する問題