上のやつを動かそうとしたとき、tensorflow-datasetsがないと動かないのですが、
conda install tensorflow-datasets
でやるとバージョンが古いので失敗します。
なので、
pip install tensorflow-datasets
でやるとうまくいきます。
余談
conda uninstall tensorflow-datasets
でやってからpipでインストールするとなぜかうまくいきませんでした。
なので、conda環境から作り直しました。
当たり前なのですが、横着してbaseに入れても色々面倒なので、やっぱり新しく仮想環境を作るのが良いですね。