最近在学习tensorflow的教程,跑cifar的例子的时候,有一句
from tensorflow.models.image.cifar10 import cifar10
跑程序的时候报错No module named 'tensorflow.models' 。之前安装的时候我是用Anaconda安装的,安装的时候没有下载models文件。
对于这个问题,可以在github上下载对应的文件。步骤如下:
1. 下载models
git clone https://github.com/tensorflow/models
Cloning into 'models'...
remote: Counting objects: 21199, done.
remote: Compressing objects: 100% (15/15), done.
remote: Total 21199 (delta 1), reused 4 (delta 0), pack-reused 21184
Receiving objects: 100% (21199/21199), 558.32 MiB | 1.15 MiB/s, done.
Resolving deltas: 100% (12433/12433), done.
Checking connectivity... done.
Checking out files: 100% (2669/2669), done.
2. 复制到tensorflow的文件夹下
如果不知道tensorflow的文件夹在哪里,可以通过以下方式查询
rin@ubuntu:~$ python
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
[GCC 5.4.0 20160609] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import tensorflow as tf
>>> tf.__path__
['/usr/local/lib/python3.5/dist-packages/tensorflow']
注意path前后各有两条下划线.
接着利用cp指令赋值。因为涉及到usr文件夹,所以要用管理员权限。需要注意的是models文件夹下面还有文件夹,需要加上-r参数递归赋值
rin@ubuntu:~$ sudo cp -r /home/rin/models /usr/local/lib/python3.5/dist-packages/tensorflow/
3. 修改路径
将原来的from tensorflow.models.image.cifar10 import cifar10,根据实际文件夹的分布改成
from tensorflow.models.tutorials.image.cifar10 import cifar10