优先阅读官方文档!!!
- 使用环境:ubuntu 16.04 + Nvidia 1060
安装tensorflow+cuda+cudnn的正确方法:
- 确定cuda和cudnn的版本
阅读tensorflow的官方文档,找到下面这张图:
该图片可以在此处找到,从中确定经过测试的兼容的cuda和cudnn的版本。
- 安装cuda
不需要单独安装nvidia的显卡驱动,因为cuda里已经包含了合适版本的显卡驱动。
在Nvidia的网站下载cuda,需要下载以前的版本去Legacy Releases里下载。详细的安装方法照着里面的文档进行。推荐使用deb(local)封装安装。 - 安装cudnn
和前面一样在官网下载,并参照官方文档安装。也推荐使用deb格式安装。 - 安装tensorflow
推荐使用pip安装
pip3 install tensorflow-gpu
完成!
网友评论