基本是按照这篇:Ubuntu18.04+RTX2080+cuda10+tensorflow安装的
链接:https://blog.csdn.net/weixin_40859436/article/details/83152249
但是出现了一些问题,
(1)在今天2018年11月25日安装的时候,这个地方选项有三项,多了一项415的驱动,选择了410.
图片来自上文博客
(2)安装tensflow(gpu版)1.5的时候没问题,跑代码的时候报错如下:
最后在同学的帮助下,改为最新的tensorflow1.12后成功运行。
(3)cudnn用了的7.3版本。
2019年1月22日
用了一段时间这个版本的tensorflow,又出现新的问题,跑keras的yolov3用1*1卷积核的时候出现了Cublas报错的问题,
遇到的错误和这个英伟达论坛里这个帖子的问题一样,只不过他是win系统上遇到这个问题,我是在ubuntu上遇到这个问题。
链接:https://devtalk.nvidia.com/default/topic/1042920/which-version-of-cuda-can-work-with-rtx-2080/
最后下载了特别版本的tensorflow之后成功解决:https://github.com/yaroslavvb/tensorflow-community-wheels/issues
这部分参考了:https://blog.csdn.net/macunshi/article/details/84749270 感谢作者的分享