首先保证自己的环境中已正确安装并切换至Nvidia显卡驱动,并已配置对应版本cuda+cudnn。
1.参考官网在虚拟环境中安装对应版本Tensorflow:
2.打开pycharm:File-Setting-Project:[name]-Project Interpreter
3.pycharm里面缺少libcublas.so.9.0
解决办法:
手动添加环境变量:
PYTHONUNBUFFERED=1;LD_LIBRARY_PATH=/usr/local/cuda-9.0/lib64;usr/local/lib
修改pycharm.sh
关闭pycharm,找到pycharm目录,打开终端执行
sudo gedit bin/pycharm.sh
在IFS="$(printf '\n\t')"
后面按Enter空出一行,然后添加
LD_LIBRARY_PATH="/usr/local/cuda/lib64:/usr/local/cuda-9.0/lib64:/usr/local/cuda-9.0/extras/CUPTI/lib64"
保存退出,重新运行pycharm,发现tensorflow可以用了