python setup.py: установить тензор потока или тензор потока-gpu?

Question

Я разрабатываю новый проект, который требует tensorflow.Поскольку TensorFlow имеет более одной установки (tensorflow и tensorflow-gpu), как мне добавить в мой раздел install_requires, что любая из них подойдет?

Answer 1

Итак, вот как я это сделал:

from pkg_resources import DistributionNotFound, get_distribution
from setuptools import setup, find_packages    

def get_dist(pkgname):
    try:
        return get_distribution(pkgname)
    except DistributionNotFound:
        return None

install_deps = ['numpy', 'tensorflow']
if get_dist('tensorflow') is None and get_dist('tensorflow-gpu') is not None:
    install_deps.remove('tensorflow')

setup(..., install_requires=install_deps)