最近刚转到Ubuntu上学习深度学习。在安装开发环境时踩了许多坑。其中一个就是在使用Pycharm导入tensorflow-gpu版时会报如下错误:
ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
Failed to load the native TensorFlow runtime.
出现的原因可能为:
- 没有安装Cuda或安装了错误的版本。在一开始我安装的是最新发布的Cuda10.0, 安装后报错。
- 直接使用桌面图标而不是Shell启动Pycharm。这样会导致Pycharm无法导入环境变量配置文件~/.bashrc。因此出现了可以在Shell中的Python环境导入但不能在Pycharm环境中导入的情况。
针对以上两种情况的解决方法为:
- 建议安装Cuda9.0版本并在~/.bashrc中设置Cuda相关的环境变量。安装教程在此不再赘述。
- 使用Shell命令: pycharm进行启动
- 如果仍想使用图标启动,则需要在Pycharm的Run->Edit Configurations界面的Environment variables列表中加入Cuda的环境变量项,具体为:
LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64
(Cuda安装路径)。在File->Setting->Build, Execution, Deployment->Console->Python Console中加入Cuda的环境变量。
相关讨论: