码迷,mamicode.com
首页 > Windows程序 > 详细

Win10安装Tensorflow的Gpu版本问题

时间:2018-06-09 23:21:40      阅读:242      评论:0      收藏:0      [点我收藏+]

标签:下载地址   AC   file   地址   com   ram   http   ade   版本问题   

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

Win10安装Tensorflow的Gpu版本问题

标签:下载地址   AC   file   地址   com   ram   http   ade   版本问题   

原文地址:https://www.cnblogs.com/tech0ne/p/9161538.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!