Tensorflow не будет импортировать с помощью sudo python3

Я работаю с удаленным экземпляром Ubuntu 18.04, где для записи файлов на python я должен запустить sudo python. Один из моих скриптов на python импортирует тензор потока, который при работе с префиксом sudo выдает следующую ошибку:

ImportError: libcublas.so.10.0: cannot open shared object file: No such file or directory

У меня была эта проблема раньше, когда у меня была установлена ​​cuda 10.1, поэтому я вернулся к cuda 10.0 и tenorflow хорошо выполняет импорт при запуске скрипта Python без команды sudo, но ошибка сохраняется, если используется sudo.

Кто-нибудь знает, в чем может быть проблема.

PS У меня есть путь cuda, добавленный к переменным среды, например:

export PATH=/usr/local/cuda-10.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
0
задан 24 May 2019 в 17:54

1 ответ

При запуске с sudo ваши PATH и LD_LIBRARY_PATH сбрасываются на (более безопасные) значения по умолчанию, поэтому ваши библиотеки CUDA больше не найдены. Вы можете сбросить их явно в вашем скрипте.

0
ответ дан 24 May 2019 в 17:54

Другие вопросы по тегам:

Похожие вопросы: