tensorflow trace: Couldn't open CUDA library libcupti.so.8.0

xiaoxiao2021-02-28  70

在使用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"

问题解决。

Reference

http://blog.csdn.net/xiaojiajia007/article/details/68483223https://github.com/tensorflow/tensorflow/issues/1824
转载请注明原文地址: https://www.6miu.com/read-70545.html

最新回复(0)