我的服务器上有一个安装了 tensorflow 的环境。它可以按预期访问 CUDA 库:
>>> import tensorflow as tf
2021-05-08 22:03:21.839382: I tensorflow/stream_executor/platform/default/dso_loader.cc:48] Successfully opened dynamic library libcudart.so.10.1
但是当我通过笔记本电脑上的远程 pycharm 激活这个环境时,tensorflow 看不到这个 CUDA 库:
import tensorflow as tf
2021-05-08 21:56:20.060856: W tensorflow/stream_executor/platform/default/dso_loader.cc:59] Could not load dynamic library 'libcudart.so.10.1'; dlerror: libcudart.so.10.1: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: '/usr/local/cuda-10.2/lib64:/usr/local/cuda-10.2/extras/CUPTI/lib64'
2021-05-08 21:56:20.060908: I tensorflow/stream_executor/cuda/cudart_stub.cc:29] Ignore above cudart dlerror if you do not have a GPU set up on your machine.
它说没有找到名为 libcudart.so.10.1
的文件,但该文件确实位于我服务器上的 /usr/local/cuda-10.2/lib64
中。