Tensorflow 无法通过远程 pycharm 访问 CUDA 库 10.1,但可以在本地服务器上访问

时间:2021-05-08 14:16:32

标签: python tensorflow pycharm gpu

我的服务器上有一个安装了 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 中。

0 个答案:

没有答案