ubuntu16.04配置tensorflow-gpu
个人记录:
- 安装nvidia驱动
- 安装cuda
- 安装cudnn
- 安装tensorflow-gpu
安装nvidia驱动
软件更新-安装之后,cuda安装驱动的地方选择no
安装cuda
参考:https://blog.csdn.net/qlulibin/article/details/78714596
安装cudnn
参考:https://www.cnblogs.com/go-better/p/7625541.html
安装tensorflow-gpu
感谢参考链接的作者
cuda9.1与tensorflow不融洽,这个大坑要躲开