找到tensorflow对应的cuda cudnn,我的环境是python3.5+Ubuntu16.04+ cuda9.0 +cudnn7.0+tensorflow-gpu 1.6 好像可以支持到1.8
1.不能在pycharm中直接安装tensorflow-gpu (cpu版本是支持的)
2. 确认cuda ,cudnn ,python环境都是好的 ,具体安装见另一篇blog https://blog.csdn.net/qq_31119155/article/details/80793006
3.在终端执行,并确定终端是可以用的
4. 终端执行 which python
5.在pycharm中 file->setting->project->Project Interpreter
6.导入刚才显示的位置:
7.可以愉快的感受tensorflow-gpu带来的强大了。