【メモ】KerasでYOLO v3 を動かすまでの設定手順
【PC環境設定】
1.下記リンクを参考に、Anacondaをインストール
www.python.jp
2.下記リンクを参考に、Anaconda promptを立ち上げ、pipをインストール
gammasoft.jp
→「ライブラリのアップデート」の、> pip install --upgrade pypdf2まで出来ればOK。
3.pipが使えるようになったら、Anaconda promptで、以下のコマンドを入力し、必要なモジュールのインストール。
・Kerasをインストール
pip install keras
・Tensorflowの1.13.1をインストール
pip install tensorflow==1.13.1
pip install tensorflow-gpu==1.13.1
4.CUDA(10.2)とcudnn(7.6.5)を以下URLを参考にインストール
qiita.com
これでPC環境の設定は完了。
【以下のようなエラーが出た時】
「 E tensorflow/stream_executor/cuda/cuda_dnn.cc:334] Could not create cudnn handle: CUDNN_STATUS_ALLOC_FAILED」
ALLOC_FAILEDであるから、GPUのメモリが足りない。
以下のURLの一番下のページのソースコードを追記する。
satabeblog.com