Tensorflow 无法使用 sudo python3 导入

Question

我正在使用远程 Ubuntu 18.04 实例，必须运行才能使用 python 写入文件sudo python。我的一个 python 脚本导入了 tensorflow，当使用sudo前缀运行时出现以下错误：

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

我以前安装 cuda 10.1 时遇到过这个问题，所以我恢复到 cuda 10.0，并且在运行不带命令的 python 脚本时 tensorflow 导入正常，但如果使用，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}}

Answer 1

使用 sudo 运行时，您的 PATH 和 LD_LIBRARY_PATH 会重置为（更安全的）默认值，因此您的 CUDA 库不再可用。您可以在脚本中明确重置它们。