tensorflow中LD_LIBRARY_PATH的问题

来源:互联网 发布:学淘宝美工多少钱 编辑:程序博客网 时间:2024/06/05 19:25

问题描述:
I tensorflow/stream_executor/dso_loader.cc:102] Couldn’t open CUDA library libcupti.so. LD_LIBRARY_PATH:
F tensorflow/core/common_runtime/gpu/cupti_wrapper.cc:57] Check failed: f != nullptr could not find cuptiActivityRegisterCallbacksin libcupti DSO; dlerror: /home/wookayin/.virtualenvs/tfdebug/local/lib/python2.7/site-packages/tensorflow/python/_pywrap_tensorflow.so: undefined symbol: cuptiActivityRegisterCallbacks

解决方案:
As said above, this solved my issue. In my case, I edited my ./bashrc file in the home directory by appending /usr/local/cuda/extras/CUPTI/lib64 to the LD_LIBRARY_PATH.

export LD_LIBRARY_PATH=CUDAHOME/lib64:LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64

参考链接:
https://github.com/tensorflow/tensorflow/issues/2626

原创粉丝点击