vs codeで画像分類のファイルが実行できない

Question

python3とtensorflowでcifar10の画像分類をしようとしています。大まかな流れは、下記のサイトを参考にしています。
https://www.sejuku.net/blog/52907

ターミナル上での環境作成は下記の2つのサイトを参考にして作りました。

https://qiita.com/tom_ato/items/bbdf4574b3ecb0048fa1

https://qiita.com/KazaKago/items/587ac1224afc2c9350f1

ターミナル上でまずhomebrewを使いpyenv,pyenv-virtualenvをインストールしました。そこでtensorflow用のpython環境(3.7.2)を作りました。そしてtensorflowはpipを使ってインストールしました。下記のコマンドです。

$ pip3 install --upgrade tensorflow

tensorflowのインストール確認は下記のコマンドで行いました。

import tensorflow as tf

hello = tf.constant('Hello World!')
s = tf.Session()
print(s.run(hello))

結果としてはHello World!が表示されて成功していました。

それから、一番はじめに貼ったサイト上にある
cifar10_train.py
cifar10_eval.py
cifar10_input.py
cifar10.py
をvs code上に書き込みcifar10_train.py をpython3で実行しました。(具体的にはvs codeでファイルを選択しMacで副ボタンを押し、ターミナルでpythonファイルを実行を押しました)
すると、下記のようなエラーが出ました。

Traceback (most recent call last):
File "/Users/shuto/Documents/python game/cifar10_train.py", line 39, in <module>
import cifar10
File "/Users/shuto/Documents/python game/cifar10.py", line 37, in <module>
import cifar10_input
File "/Users/shuto/Documents/python game/cifar10_input.py", line 23, in <module>
import tensorflow_datasets as tfds
ModuleNotFoundError: No module named 'tensorflow_datasets'

上記サイトの手順通りに進めているのですが、原因も解決方法もわかりません。教えてください。

使用pc : macbook air2017
使用言語: python3.7.2

Comments

[hayataka2049]

環境の構築で行った手順と、実行している方法をできるだけ詳細に書いていただけると回答しやすいです。
あと、そもそもインストールがうまくいっているのかと、インストールした環境で実行しているのかを切り分ける必要があります。そのためにはtensorflowのインストール先と、実際に実行しているpythonのインストール先の情報が必要です。

[chainsmoker]

すいません、実際に実行しているpythonのインストール先の情報とは具体的にはどのような情報のことでしょうか？

[chainsmoker]

一度わかる範囲で編集しました。確認お願いします。足りないところがありましたら、また編集依頼をお願いします。

Answer 1

おそらく別途tensorflow-datasetsを導入する必要があるでしょう。

pip install tensorflow-datasets

pyenvで仮想環境を作ったのであれば、同じ環境に導入する必要があるので、その点だけ注意してください。

Comments

[chainsmoker]

回答ありがとうございました。導入すると成功しました。本当にありがとうございます。