Назад | Перейти на главную страницу

Запуск кода тензорного потока в задании крутящего момента

У меня работает кластер с крутящим моментом для распределения заданий. Я хочу запустить задание с кодом тензорного потока, и у меня возникают проблемы с тем, что тензорный поток не распознается.

Я установил тензорный поток для своего пользователя LDAP с помощью anaconda, поэтому я могу войти в среду тензорного потока на любом узле и запустить свой код вручную. Моя проблема в том, что задание крутящего момента не открывает среду при запуске, поэтому я получаю «ImportError: Нет модуля с именем tensorflow», и мой код не запускается.

Как я могу сказать узлам, что нужно запустить мой файл python в среде tenorflow?

Так выглядит мой файл задания крутящего момента

Примечание. Здесь я попытался запустить команду, открывающую среду, в других версиях я этого не делал.

Заранее благодарим за любую доступную помощь.

Извините, я забыл ответить, когда получил ответ. Если кто-нибудь увидит это в будущем, исправление моей проблемы заключалось в экспорте переменной PATH в папку bin anaconda (где он мог найти двоичный файл python, который использует anaconda и который может получить доступ тензорный поток):

export PATH="<path_to_anaconda_folder>/anaconda3/bin:$PATH"

Спасибо Tux_DEV_NULL за помощь!

Также я добавил следующее на всякий случай, чтобы избежать проблем с cuda в будущем (я использую графический процессор), я на самом деле не знаю, нужно ли это, но, возможно, это поможет кому-то ..:

export PATH=$PATH:/usr/local/cuda-8.0/bin