No module named 'tensorflow.models' 的问题解决方法

版权声明:本文为博主原创文章,遵循 CC 4.0 by-sa 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/RineZ/article/details/81671382

最近在学习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

 

展开阅读全文

没有更多推荐了,返回首页