tensorflow下载mnist数据时出错

tensorflow使用代码

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import gzip
import os
import tempfile

import numpy
from six.moves import urllib
from six.moves import xrange  # pylint: disable=redefined-builtin
import tensorflow as tf
from tensorflow.contrib.learn.python.learn.datasets.mnist import read_data_set

mnist = input_data.read_data_sets("MNIST_data/", one_hot=True)

导入mnist数据时出错 Import Error:No module name distrib

原因是tensorflow安装不全，使用以下语句更新即可

pip install -U tensorflow