Win10安装Tensorflow的Gpu版本问题

Cuda,Cudnn 和 Tensorflow gpu这三个东西要兼容

  1. 先下载最新的Cudnn,和Cuda的兼容性在下载页上有(需要注册):

    Cudnn下载地址

  2. 下载安装Cudnn的对应Cuda版本,然后将Cudnn解压拷贝到Cuda安装目录:

    Cuda下载地址

    C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\[VER.]

  3. 上面两个都下载最新相兼容的版本,然后Tensorflow gpu也更新为最新版一般就没问题了:

    pip install --upgrade tensorflow-gpu

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

推荐阅读更多精彩内容