tensorflow+cuda+cudnn 安装最佳实践

优先阅读官方文档！！！

  • 使用环境：ubuntu 16.04 + Nvidia 1060

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

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

经过测试的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

完成！

©著作权归作者所有,转载或内容合作请联系作者
平台声明：文章内容（如有图片或视频亦包括在内）由作者上传并发布，文章内容仅代表作者本人观点，简书系信息发布平台，仅提供信息存储服务。

推荐阅读更多精彩内容