tensorflow gpu从systemd服务初始化失败

Question

我正在尝试托管使用tensorflow库的flask API。我安装了带有CUDA和cudnn库的tensorflow gpu库。我使用以下命令手动检查了它是否正常。

/captcha/env/bin/gunicorn captcha:app -b 0.0.0.0:5124 -k gevent --worker-connections 1000

但是当我添加这个systemd服务时，我得到了一个tensorflow gpu错误

系统服务

[Unit]
Description=gunicorn daemon
Requires=gunicorn.socket
After=network.target

[Service]
PIDFile=/run/gunicorn/pid
User=root
Group=root
WorkingDirectory=/captcha/env
ExecStart=/captcha/env/bin/gunicorn captcha:app -b 0.0.0.0:5124 -k gevent --worker-connections 1000
ExecReload=/bin/kill -s HUP $MAINPID
ExecStop=/bin/kill -s TERM $MAINPID
PrivateTmp=true

[Install]
WantedBy=multi-user.target

日志文件中的错误文本：

无法加载本地TensorFlow运行时。

请参阅https://www.tensorflow.org/install/install_sources#common_installation_problems

出于某些常见原因和解决方案。 寻求帮助时，在此错误消息上方包括整个堆栈跟踪。

谁能指出我做错了什么？

Answer 1

我正在使用PM2来管理tensorflow-flask api进程。

http://pm2.keymetrics.io/

我使用以下命令作为内容创建外壳文件。

pm2 start run.sh

Answer 2

Systemd似乎剥离了所有环境变量，而TensorFlow需要知道在哪里可以找到Cuda。 没有LD_LIBRARY_PATH，它将失败。

可能有几种方法可以执行此操作，但这对我有用。

[Service]
Environment=LD_LIBRARY_PATH=/usr/local/cuda/lib64
ExecStart=/path/to/your/app
...