本文介绍了在使用TensorFlow进行GPU追踪时遇到的问题及解决方案。主要问题是由于缺少必要的CUDA库文件导致追踪功能无法正常工作。文章提供了如何正确配置环境变量LD_LIBRARY_PATH的方法，以确保能够加载所需的libcupti.so库。
在使用tensorflow的trace功能时报错：
2017-05-07 02:08:31.473421: I tensorflow/stream_executor/dso_loader.cc:129] Couldn’t open CUDA library libcupti.so.8.0. LD_LIBRARY_PATH:
原因 found at https://github.com/tensorflow/tensorflow/issues/1824 ：
Note: to get GPU tracing you will need to ensure that libcupti.so is on you LD_LIBRARY_PATH. It is usually found in /usr/local/cuda/extras/lib64.
所以需要配置LD_LIBRARY_PATH
export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:/usr/local/cuda-8.0/extras/CUPTI/lib64"问题解决。
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
