python - Tensorflow:在 setup.py 中自动安装

Question

在 setup.py 文件中，我们通常会放一些类似的东西

install_requires=["numpy>=1.11.2", "scipy>=0.17.1"]

以便在运行命令 python setup.py install 时，会自动安装依赖项。

但是，我无法让它与 tensorflow 一起使用。我尝试将 tensorflow>=1.0.1 或 tensorflow-gpu>=1.0.1 和一堆其他组合，但我总是收到相同的错误消息:

错误:找不到适合 Requirement.parse('tensorflow>=1.0.1') 的分布

关于如何解决这个问题有什么想法吗？

请注意，我可以使用 pip install tensorflow 或 pip install tensorflow-gpu 安装 tensorflow。

Answer 1

为什么不在 setup.py 中使用 pip？

像这样:

import pip

def install(packagename):
    pip.main(['install', packagename])

install('tensorflow')

分别放到另一个setup调用的脚本中。