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

请添加图片描述

Logo

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐