pycharm中的导入tensorflow时出现cuda路径库问题,首先cuda9.0,cudnn7.0都已经正确安装,并可以正常运行。在终端运行程序时没有问题。但就是在pycharm中出现问题。如下图
解决方法如下:亲测有效
step 1:打开 Run-->Edit Configurations,按下图输入环境变量的路径,实际的路径根据自己的环境做相应的替换
CUDA_HOME /usr/local/cuda-9.0/
LD_LIBRARY_PATH /usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
step 2:File--->Setting,如下图所示,按照图中,在添加环境变量中添加cuda路径
CUDA_HOME /usr/local/cuda-9.0/
LD_LIBRARY_PATH /usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64:$LD_LIBRARY_PATH
设置完以后,重新启动pycharm,发现还是出现原来的问题。(我是点击Ubuntu左边的pycharm快捷键启动)
此时,就需要换一种启动方式,直接从pycharm的安装路径里启动,命令如下:进入到pycharm的安装文件里的bin。执行./pycharm.sh,启动pycharm,
cd pycharm
cd bin
./pycharm.sh
运行程序发现,问题解决了,没有出现cuda的路径问题,但为什么从快捷键启动会出现cuda路径问题呢,原因是从桌面快捷键启动的路径与从pycharm启动的路径不同,并未包含cuda的路径。需要把cuda的路径添加到profile中,解决方案如下,通过如下命令打开profile:
sudo gedit /etc/profile
打开后,在最后面添加如下两句路径:
export PATH="/usr/local/cuda-9.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"
保存后,重启电脑,重新通过快捷键启动pycharm,运行import tensorflow as tf ,导入成功,问题解决!
注意:以上的cuda路径,需要根据具体安装的cuda的路径,以及不同的版本做相应的修改。