pycharm不能调用tensorflow-gpu的问题
在终端import tensorflow没有问题↓
但是在pycharm中导入却不能使用GPU
报错:
W tensorflow/stream_executor/platform/default/dso_loader.cc:64] Could not load dynamic library 'libcudart.so.11.0'; dlerror: libcudart.so.11.0: cannot open shared object file: No such file or directory
原因是本地pycharm编译器找不到虚拟环境中的cuda的信息
,因此在pycharm 中添加此信息:
选择pycharm 菜单栏Run ——> Run-Edit Configurations ——> Environment variables——> 将cuda的路径加进去 例如:LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64