意外となんとかなる日記

世迷言を心に、綺麗事を頭に

スポンサーリンク

Windows10(1709)のRTX2080Tiでmnist_cnn.pyを動作させるのにtensorflow-gpuとkerasのバージョンで困った話

 f:id:Suechan:20190909135934j:plain

 久しぶりにちょっとPythonを動かしてみよっかなーとか思ってインストールしただけなのに、見事にハマってしまいますた。

 

 以前書いたように、Windows 10でtensorflow-gpuをインストールする手順とかを見ながら普通に、

pip install tensorflow-gpu keras

 とかしたら、よくわからないけどmnist_cnn.pyが動作しないという事象に見舞われました。

 どうもtensorflow-gpuとかnumpyとかkerasとかのライブラリのバージョンがアカンっぽいです。

 というところでイロイロとライブラリのバージョンの調査とかをしてなんとかなったので、ワーニングを吐くがとりあえず動くまでの手順とかバージョンとか遺します。

 

 そもそもはこのソースは古いバージョンだから、ワーニングが出るくらいは当たり前という前提です。

 

 以下、端的にやったことを書いていきます。

 

 1.Windows10新規インストール

  諸事情により1709を使いましたが、1903とかだとまたなんか違うのかどうだろうか・・・。

 

 2.C++2015インストール

  なんかかち合いがちなので、先にインストールします。

 

 3.NVIDIAドライバインストール

  ここは1709特有で419.67を使いました。RTX2080Tiとか

  1903とかだと43X.XXとかになりそうですが、Game ReadyかStudioかわからんです・・・。

 

 4.CUDA 10.0.130インストール

 5.cudnn 10.0 v7.5.0.56インストール(ファイルコピー)

 6.Python 3.6.8インストール

  とりあえずこの組み合わせで動きます。詳細な手順は前のリンク先とかノリと勢いとかで。

  GTX10XX系ならCUDA 9.0.176&cudnn 9.0 v7.3.1.20辺りかと思われます。

 

 7.コマンドプロンプト(管理者)起動

  pipかnumpyのアップデートがうまくいかないので、管理者で起動します。

 

8.環境作成

python -m venv E:\python368\work

E:

cd E:\python368\work\

Scripts\activate

 

  9.各ライブラリをバージョン指定してインストール

  こちらはRTX20XX用ですので、GTX10XX系だと(CUDAを含めて)もう少しバージョンを下げれると思います。このバージョンを変えると、実行時にエラーとなりがちです(ある程度確認済み

python -m pip install --upgrade pip

pip install setuptools==41.2.0 numpy==1.16.1 tensorflow-gpu==1.14.0 keras==2.1.6

 

 pip listの結果です。

(work) E:\Python368\work>pip list

Package             Version

-------------------- -------

absl-py             0.8.0

astor               0.8.0

gast                 0.3.1

google-pasta         0.1.7

grpcio               1.23.0

h5py                 2.10.0

Keras               2.1.6

Keras-Applications   1.0.8

Keras-Preprocessing 1.1.0

Markdown             3.1.1

numpy               1.16.1

pip                 19.2.3

protobuf             3.9.1

PyYAML               5.1.2

scipy               1.3.1

setuptools           41.2.0

six                 1.12.0

tensorboard         1.14.0

tensorflow-estimator 1.14.0

tensorflow-gpu       1.14.0

termcolor           1.1.0

Werkzeug             0.15.6

wheel               0.33.6

wrapt               1.11.2

 

 10.mnist_cnn.py実行

  実行結果がこちらです。

f:id:Suechan:20190909143546j:plain

  イロイロワーニングを吐いてますが、どうも「その書き方はそのうちアカンでー」みたいな感じっぽい?

 

 以上が、とりあえず動くまでの手順書となります。

 

 ここまで調べるのに丸一日とか使いましたです。

 前のときにpip listしとけば、こんなことには・・・。

スポンサーリンク