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 и тензор потока импортируется нормально при запуске скрипта 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}}

1 ответ

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

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