Tensorflow のアップデート(2.4.1 -> 2.6.0)

IT

家のPCのTensorflowを2.4.1から2.6.0にアップデートした。

手順は、ほぼ前回どおり。

guppi524.hatenablog.jp

今回も、いつものごとく、下記のサイトの手順にお世話になる。ありがとうございます。

www.kkaneko.jp

インストールしたツールは、

  • CUDA 11.4 Update2
  • cuDNN 8.2.4
  • Tensorflow 2.6.0 (pipパッケージ)
    • Keras 2.6.0 (Tensorflow pipパッケージ組み込み)

Tensorflow 2.6.0から、kerasが統合されているっぽい

newreleases.io

github.com

なので、

pip install tensorflow

とすると、kerasも一緒にインストールされた。

Successfully installed clang-5.0 gast-0.4.0 grpcio-1.40.0 h5py-3.1.0 keras-2.6.0 tensorboard-2.6.0 tensorboard-data-server-0.6.1 tensorflow-2.6.0 tensorflow-estimator-2.6.0

動作自体は何事もなく動いたが、kerasのパッケージングまわりが変わっていると思われるため、下記の2つのインポートでエラーがでた。

from keras.optimizers import SGD
from keras.utils import plot_model

で、

from tensorflow.keras.optimizers import SGD
from tensorflow.keras.utils import plot_model

問題なく動作することを確認。

学習(model.fit)の速度は、2.4.1のときとそれほど変わらない感じでした。CIFAR-10 画像分類10エポックの学習時間は、以下のとおり。

2.6.0: 実行時間:125.13618564605713[秒]
2.4.1: 実行時間:128.62278509140015[秒]

ベンチマークに使用したモデルはこちら。 github.com