テクめも

プログラミング関連のちょっとしたTipsなどを書いています。

Tensorflow

TensorFlowとPython/CUDAのバージョン対応表

TensorFlow Python CUDA cuDNN 1.14.0 3.3~3.7 10 7.4 1.13.1 3.3~3.7 10 7.4 1.12.0 3.3~3.6 9 7 1.11.0 3.3~3.6 9 7 参考 Build from source | TensorFlow

TensorflowのSaved Modelを確認する

Tensorflowをインストールすると同時にインストールされるsaved_model_cliを利用すると、Saved Modelの入出力を確認することができます。 $ saved_model_cli show --dir models/mnist/1/ --all MetaGraphDef with tag-set: 'serve' contains the following S…

KerasでLSTMが使えないときの対処法

KerasでLSTMを使おうとしたとき以下のようなエラーが発生しました。 TypeError: while_loop() got an unexpected keyword argument 'maximum_iterations' 調べてみると、バージョンがよくなかったみたいで、kerasのバージョンを変更するとうまく動きました。…

TensorflowのImportErrorを解決する

GPU環境でTensorflowを実行したとき ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory というエラーが発生した場合、CUDAのライブラリのパスが通っていないことに原因があります。 export LD_LIBRARY_PATH=/usr/l…