tensorflow+cuda+cudnn 安装最佳实践

优先阅读官方文档!!!

  • 使用环境:ubuntu 16.04 + Nvidia 1060

安装tensorflow+cuda+cudnn的正确方法:

  1. 确定cuda和cudnn的版本
    阅读tensorflow的官方文档,找到下面这张图:

tensorflow+cuda+cudnn 安装最佳实践_第1张图片
经过测试的tf,cuda,cudnn的版本

该图片可以在 此处找到,从中确定经过测试的兼容的cuda和cudnn的版本。

  1. 安装cuda
    不需要单独安装nvidia的显卡驱动,因为cuda里已经包含了合适版本的显卡驱动。
    在Nvidia的网站下载cuda,需要下载以前的版本去Legacy Releases里下载。详细的安装方法照着里面的文档进行。推荐使用deb(local)封装安装。
  2. 安装cudnn
    和前面一样在官网下载,并参照官方文档安装。也推荐使用deb格式安装。
  3. 安装tensorflow
    推荐使用pip安装
    pip3 install tensorflow-gpu

完成!

你可能感兴趣的:(tensorflow+cuda+cudnn 安装最佳实践)