美文网首页
tensorflow trace: Couldn't open

tensorflow trace: Couldn't open

作者: 我的章鱼小丸子呢 | 来源:发表于2020-07-23 19:33 被阅读0次

问题:在使用tensorflow的trace功能时报错:

找不到CUDA library libcupti.so

原因:LD_LIBRARY_PATH路径下找不到这个包

官方说明: https://github.com/tensorflow/tensorflow/issues/1824

Node: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.

解决方法:找到虚拟环境/本机cuda的lib文件夹,添加到环境变量LD_LIBRARY_PATH.

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

相关文章

网友评论

      本文标题:tensorflow trace: Couldn't open

      本文链接:https://www.haomeiwen.com/subject/jxtklktx.html