首页 > 解决方案 > Tensorflow 无法通过远程 pycharm 访问 CUDA 库 10.1,但可以在本地服务器上访问

问题描述

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

标签: pythontensorflowpycharmgpu

解决方案


推荐阅读