2017-03-28 34 views
9

W pliku setup.py zwykle umieścić cośTensorflow: zainstalować go automatycznie w setup.py

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

tak, że gdy komenda python setup.py install uruchomieniu Zależności są instalowane automatycznie.

Jednak nie udaje mi się sprawić, żeby działało z tensorflow. Starałem się umieścić tensorflow>=1.0.1 lub tensorflow-gpu>=1.0.1 i kilka innych kombinacji ale sprecyzowane uzyskać ten sam komunikat o błędzie:

error: Could not find suitable distribution for Requirement.parse('tensorflow>=1.0.1')

Każdy pomysł na jak rozwiązać ten problem?

Zauważ, że nie mam problemu z instalacją tensorflow przy użyciu pip install tensorflow lub pip install tensorflow-gpu.

+0

Co się stanie, jeśli pominiesz '> = 1.0.1'? Czy to działa? Czy ostatnio zaktualizowałeś 'pip':' pip install --upgrade pip'? Która wersja jest zainstalowana z 'pip install tensorflow' lub' pip install tensorflow-gpu'? – Dair

+0

@Dair Próbowałem również bez '> = 1.0.1' i wcześniej uaktualniłem mój pip. Zainstalowana wersja to '0.12'. – BiBi

+0

Ponadto, jeśli TF jest już zainstalowany, można go wykryć tylko, jeśli poprawna wersja jest wymieniona w 'install_requires', tj. Wprowadzenie' tensorflow' w 'install_requires' nie może wykryć' tensorflow-gpu' i na odwrót. Jakiekolwiek obejście tego? – optimist

Odpowiedz

0

dlaczego nie używasz pip w swoim setup.py?

Jak to:

import pip 

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

install('tensorflow') 

Odpowiednio umieścić go na inny skrypt o nazwie przez setup.