Ubuntu16.04 Pycharm中import tensorflow 报错
环境配置:
- Ubuntu16.04
- Tensorflow-gpu 1.9.0
- CUDA 9.0
- CUDNN 7.1
问题描述:
关于tensorflow-gpu的版本配置一切正确,import tensorflow时报错。
> ImportError: libcublas.so.9.0: cannot open shared object file: No such file
请先保证已经正确安装tensorflow-gpu对应的CUDA及CUDNN版本!!!
解决办法:
错误提示不存在CUDA 9.0的相关文件。
其实并不是未安装,而是系统路径出现问题。
从终端命令行启动pycharm
使用cd命令进入pycharm的安装文件夹中的bin文件夹
执行
./pycharm.sh
之后进入pycharm编程界面,import tensorflow未报错。
问题解决。
但是再次从桌面图标进入pycharm仍会报错。
这是因为从桌面图标点击进入pycharm,系统路径与终端命令行中的有所不同。
在终端命令行中输入:
echo $PATH
存在CUDA 9.0的系统路径。
从桌面图标启动进入pycharm后,编写测试文件,运行如下代码。
import os
os.system("echo $PATH")
不存在CUDA 9.0的系统路径
Ubuntu的系统路径配置的地方一般是 /etc/profile 和 ~/.bashrc 两个文件。
本人将CUDA 9.0的路径加入到~/.bashrc中。
然而对比之下发现:
从桌面图标启动的pycharm无法检测到~/.bashrc中的$PATH
故将CUDA 9.0的路径加入到 /etc/profile文件中。
sudo vi /etc/profile
在文件末尾加入以下语句
# added by cuda_9.0 installer
export PATH="/your_path_to/cuda-9.0/bin:$PATH"
export LD_LIBRARY_PATH="/your_path_to/cuda-9.0/lib64:$LD_LIBRARY_PATH"
保存,重启系统后,问题得到真正解决。
参考资料: