https://zhuanlan.zhihu.com/p/51008895
如果想要在pycharm正确运行tensorflow-gpu版本,需要为pycharm设置与终端一样的gpu环境。
方法1:
在pycharm中针对项目设置环境变量,如下图所示:
右上角-->Edit Configurations-->Environment Variables-->添加cuda的lib目录(在~/.bashrc中找 LD_LIBRARY_PATH=usr/local/cuda-9.0/lib64),如下图所示:
这种方法,麻烦在于,每创建一个tensorflow项目,需要设置一下pycham的环境变量。
方法2:
通过终端打开的pycharm会继承.bashrc中设置的环境变量。比如,我的pycharm安装目录在/opt/software/pycharm下,执行/opt/software/pycharm/bin目录下的pycharm.sh脚本打开pycharm就不会出问题了。
如果你觉得这种打开方式麻烦,可以建立一个软连接,以后每次就可以使用pycharm命令打开了pycharm编辑器了:
sudo ln -s /opt/software/pycharm/bin/pycharm.sh /usr/local/bin/pycharm
---参考资料:
https://blog.csdn.net/DL_ChenBo/article/details/53262230