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

1. 问题

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

# python mnist.py

2017-08-24 08:57:37.175475: tensorflow/stream_executor/dso_loader.cc:129
Could 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_PATH
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"

# source ~/.bash_profile