从setup.py安装时，无法在Google Colab中导入Tensorflow 2.2.0rc2

Question

我试图在谷歌Colab中导入最新的rc2版本的Tensorflow（目前为2.2.0rc2），但是从我的 setup.py 安装脚本安装时无法做到这一点。

当我使用手动安装Tensorflow时！pip从Colab单元安装tensorflow==2.2.0rc2，一切正常，我可以导入tensorflow。

接下来是我如何在Google Colab中安装依赖项安装设置：

# Executes the cell in bash mode
%%bash

if [ ! -d "/content/deep-deblurring/" ]; 
    then 
        git clone https://github.com/ElPapi42/deep-deblurring;
        cd deep-deblurring/
    else 
        cd deep-deblurring/; 
        git pull; 
fi;

git checkout development
cd ..

pip uninstall -y tensorflow tensor2tensor tensorboard tensorboardcolab tensorflow-datasets tensorflow-estimator tensorflow-gan tensorflow-hub tensorflow-metadata tensorflow-privacy tensorflow-probability

pip install colab-env
pip install --upgrade grpcio

cd deep-deblurring/
python setup.py install
cd ..

接下来是我的 setup.py 文件：

#!/usr/bin/python
# coding=utf-8

"""Setup and install the package and all the dependencies."""

from setuptools import setup, find_packages

with open('requirements.txt') as pro:
    INSTALL_REQUIRES = pro.read().split('\n')

setup(
    author='Whitman Bohorquez, Mo Rebaie',
    author_email='[e-mail]',
    name='deblurrer',
    license='MIT',
    description='Image Deblurring using Deep Learning Architecture',
    version='1.0.0',
    url='',
    packages=find_packages(),
    include_package_data=True,
    python_requires='>=3.6',
    install_requires=INSTALL_REQUIRES,
    classifiers=[
        'Development Status :: Alpha',
        'Programming Language :: Python',
        'Programming Language :: Python :: 3.6',
        'Intended Audience :: Developers',
    ],
)

其次是要求。txt文件：

grpcio == 1.27.2
kaggle
numpy
tensorflow >= 2.2.0rc2
pandas

事实上，Google Colab附带Tensorflow 2.2.0rc1，但我想要rc2.当我执行：

import tensorflow as tf
print(tf.__version__)

在执行setup.py安装脚本之前，导入正常工作。但是在使用setup.py安装完成后，错误导入错误：没有名为“张量流”的模块被抛出。

我在python安装执行之前和之后检查了tensorflow setup.py 安装，一切似乎都很好，安装前使用tensorflow 2.2.0rc1，安装后使用2.2.0rc2。

正如我首先提到的，当我手动使用安装Tensorflow时！pip安装Tensorflow==2.2.0rc2导入按预期工作，所以问题必须围绕setup.py文件或需求，类似的东西，但我没有看到它。

希望你们的帮助！

PD:这个项目设置在上周五还在运行，但是今天我试着运行它，突然没有明显的原因就停止了。

PD2：https://colab.research.google.com/drive/1Qv8h4ceEtDTq5lvt1uKJG8dk53_bUqBZ 这是我与您共享的 Colab 笔记本，这将设置重现问题的代码。

PD3:这是导入tensorflow时Google Colab中的完整错误回溯抛出:

---------------------------------------------------------------------------
AttributeError                            Traceback (most recent call last)
/usr/lib/python3.6/importlib/_bootstrap.py in _find_spec(name, path, target)

AttributeError: '_TensorflowImportHook' object has no attribute 'find_spec'

During handling of the above exception, another exception occurred:

ImportError                               Traceback (most recent call last)
2 frames
<ipython-input-7-69e5d056d1fc> in <module>()
----> 1 import tensorflow as tf
      2 
      3 tf.__version__

/usr/local/lib/python3.6/dist-packages/google/colab/_import_hooks/_tensorflow.py in find_module(self, fullname, path)
     26     if fullname != 'tensorflow':
     27       return None
---> 28     self.module_info = imp.find_module(fullname.split('.')[-1], path)
     29     return self
     30 

/usr/lib/python3.6/imp.py in find_module(name, path)
    295         break  # Break out of outer loop when breaking out of inner loop.
    296     else:
--> 297         raise ImportError(_ERR_MSG.format(name), name=name)
    298 
    299     encoding = None

ImportError: No module named 'tensorflow'

---------------------------------------------------------------------------
NOTE: If your import is failing due to a missing package, you can
manually install dependencies using either !pip or !apt.

To view examples of installing some common dependencies, click the
"Open Examples" button below.
---------------------------------------------------------------------------

Answer 1

我找到了一个解决方法，但到目前为止这不是这个问题的解决方案，所以这不会被接受为解决方案，但会帮助有同样麻烦的人继续他们的工作：

在安装自定义软件包之前手动安装您的需求，就我而言，这是 pip install -r “/content/deep-deblurring/requirements.txt”：

# Executes the cell in bash mode
%%bash

if [ ! -d "/content/deep-deblurring/" ]; 
    then 
        git clone https://github.com/ElPapi42/deep-deblurring;
        cd deep-deblurring/
    else 
        cd deep-deblurring/; 
        git pull; 
fi;

git checkout development
cd ..

pip uninstall -y tensorflow tensor2tensor tensorboard tensorboardcolab tensorflow-datasets tensorflow-estimator tensorflow-gan tensorflow-hub tensorflow-metadata tensorflow-privacy tensorflow-probability

pip install colab-env
pip install --upgrade grpcio

pip install -r "/content/deep-deblurring/requirements.txt"

cd deep-deblurring/
python setup.py install
cd ..

目前这修复了进口问题，但这不是一个干净的解决方案，让我们希望以后有更好的解释！

Answer 2

tenorflow没什么问题，但Colab的_TensorflowImportHook缺少find_specimpl，因此如果tenorflow安装为蛋目录，它将引发。由于钩子除了发出将tenorflow更新到2.0的通知之外没有任何有用的作用，并且无论如何都计划删除，因此一个简单的修复方法是将其从sys.meta_path笔记本开头的某个地方清除：

[1] import sys
    sys.meta_path[:] = [hook for hook in sys.meta_path if not h.__class__.__name__ == '_TensorflowImportHook']

[2] import tensorflow as tf
    print(tf.__version__)