前提・実現したいこと
みなさんこんにちは。
私は現在、Tensorflowについて勉強するために、Tensorflow-GPUとKerasをUbuntu16.04にインストールしたいと考えています。
Tensorflow-gpuインストール手順 等のサイトを参考に環境を構築しましたが、Kerasのサンプルプログラムでエラーが発生します。
エラー内容は「Tesorflowのセッションが張れない」といったものですが、Tensorflowの詳細を知らないため、何がいけないかわかりません。
ご存知の方がいらっしゃいましたら、お手数ですがご教授願います。
現在チャレンジしている構成は以下となります。
- OS : Ubuntu 16.04.4 LTS
- Kernel : GNU/Linux 4.4.0-130-generic x86_64
- GPU : TeslaM60
- Python : 3.5
- Keras : 2.1.4
- Tensorlow-gpu : 1.4.1
- cuda : 8.0.61-1
- cuDNN : 6.0.21
使用したパッケージ(NVIDIA系)
- cuda-repo-ubuntu1604_8.0.61-1_amd64.deb
- libcudnn6_6.0.21-1+cuda8.0_amd64.deb
- libcudnn6-dev_6.0.21-1+cuda8.0_amd64.deb
- libcudnn6-doc_6.0.21-1+cuda8.0_amd64.deb
使用したテストプログラム
- mnist_cnn.py(Githubより入手:https://github.com/fchollet/keras.git)
発生している問題・エラーメッセージ
ubuntu@keras:~/workspace/keras/examples$ python3 mnist_cnn.py Using TensorFlow backend. x_train shape: (60000, 28, 28, 1) 60000 train samples 10000 test samples Train on 60000 samples, validate on 10000 samples Epoch 1/12 2018-07-03 21:39:59.483182: E tensorflow/core/common_runtime/direct_session.cc:170] Internal: failed initializing StreamExecutor for CUDA device ordinal 0: Internal: failed call to cuDevicePrimaryCtxRetain: CUDA_ERROR_NOT_SUPPORTED Traceback (most recent call last): File "mnist_cnn.py", line 67, in <module> validation_data=(x_test, y_test)) File "/home/ubuntu/.local/lib/python3.5/site-packages/keras/models.py", line 963, in fit validation_steps=validation_steps) File "/home/ubuntu/.local/lib/python3.5/site-packages/keras/engine/training.py", line 1712, in fit validation_steps=validation_steps) File "/home/ubuntu/.local/lib/python3.5/site-packages/keras/engine/training.py", line 1235, in _fit_loop outs = f(ins_batch) File "/home/ubuntu/.local/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py", line 2473, in __call__ session = get_session() File "/home/ubuntu/.local/lib/python3.5/site-packages/keras/backend/tensorflow_backend.py", line 176, in get_session _SESSION = tf.Session(config=config) File "/home/ubuntu/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 1482, in __init__ super(Session, self).__init__(target, graph, config=config) File "/home/ubuntu/.local/lib/python3.5/site-packages/tensorflow/python/client/session.py", line 622, in __init__ self._session = tf_session.TF_NewDeprecatedSession(opts, status) File "/home/ubuntu/.local/lib/python3.5/site-packages/tensorflow/python/framework/errors_impl.py", line 473, in __exit__ c_api.TF_GetCode(self.status.status)) tensorflow.python.framework.errors_impl.InternalError: Failed to create session.
補足情報(FW/ツールのバージョンなど)
最終的には目標サイト のように動画解析を行いたいと考えています。
あなたの回答
tips
プレビュー