在 tensorflow 中安装 models 模块

  • 在学习 rnn 时,遇到一个小问题,引入 models 模块时(也就是下面这一行代码)出错,
from tensorflow.models.rnn.ptb import reader

报错 No module of tensorflow.models

  • google 一下,了解到 tensorflow 将 models 模块当做第三方库来使用,需要自己手动安装

  • 接下来,如何安装呢,首先找到自己的 tensorflow 文件夹,(声明,我用的是 anaconda开发工具,有一个名为 tensorflow 开发环境,里面安装有 tensorflow 版本 1.8.0)

  • 找路径,费劲啊,网上我是没找到关于 anaconda 的,然后总结经验,左思右想,一步一坑,终于找到了,我的 tensorflow 环境下的 tensorflow 文件夹路径为
    /home/gcb/anaconda3/envs/tensorflow/lib/python3.6/site-packages/tensorflow

  • 然后用命令安装 models 模块

git clone --recurse-submodules https://github.com/tensorflow/models
  • 安装好之后,可以看到 tensorfow 文件夹下,有models 模块了


  • 接下来呢,再次运行 from tensorflow.models.rnn.ptb import reader
    又报错,什么 No module of tensorflow.models.rnn
    经查看,models 文件夹下面没有 rnn 文件夹


    怎么回事呢,原来 rnn 文件夹,跑到tutorials 文件夹下面了

  • 好吧,再次修改命令,加上目录 tutorials 就好

from tensorflow.models.tutorials.rnn.ptb import reader
  • 再次运行,啊,还出错,咋办呢,说 reader 有问题,好吧
    打开 /models/tutorials/rnn/ptb/init.py 文件,然后将
import reader
import util

修改为

from tensorflow.models.tutorials.rnn.ptb import reader
from tensorflow.models.tutorials.rnn.ptb import util

OK,大功告成,可以愉快地 Coding 了!!!

  • 参考

你可能感兴趣的:(在 tensorflow 中安装 models 模块)