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和其他一些组合放在一起，但我总是得到相同的错误消息：

error: Could not find suitable distribution for 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')

分别将其放入设置调用的另一个脚本中。

Answer 2

到目前为止，它可以工作。在install_requires=["tensorflow>=1.12"]中指定setup.py并执行python setup.py install将安装所需的张量流版本。

它也可以与tensorflow-gpu一起使用。