tensorflow运维（3）：tensorflow1.0的一个bug：can't open CUDA library libcupti.so.8.0

来源：互联网发布：农业物联网大数据平台编辑：程序博客网时间：2024/05/22 10:36

1. 问题

今天，我在centos7上使用tensorflow训练模型时，出现以下报错：

# python mnist.py2017-08-24 08:57:37.175475: tensorflow/stream_executor/dso_loader.cc:129Could not open CUDA library libcupti.so.8.0. LD_LIBRARY_PATH: :/usr/local/cuda/lib64:/usr/local/cuda-8.0/lib64

原因是新旧tensorflow版本默认寻找libcuti.so.8.0的路径改变了，或者存放相应文件的目录改变了。因此，我们需要配置环境变量LD_LIBRARY_PATH。

2. 解决方法

法1. 一次生效

在命令行，键入以下指令：

# export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"

法2. 永久生效

在文件中，加入环境变量：

# vim /etc/profile      //如果是普通用户， vim ~/.bash_profile //配置环境变量LD_LIBRARY_PATHexport LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"# source ~/.bash_profile

阅读全文

0 0