windows10安装tensorflow-gpu1.3.0

计算机系统和显卡配置

本机系统windows10 64位、显卡英伟达 GTX1060

安装tensorflow的版本选择

注意:这里需要特备留意cudnn的版本必须与tensorflow-gpu的版本匹配,tensorflow的版本为1.3.0。Anaconda安装后可使用命令conda list查看。

开始安装

(1)安装Anaconda

使用pip -V查看版本,低于9.0则需要输入python -m pip install -U pip更新。

推荐大家使用清华源镜像,这样以后使用起来比较方便,也避免了一些下载速度安装出错的各种问题。在命令提示符里输入:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/

conda config --set show_channel_urls yes (当前cmd配置conda)

(2)安装cuda

点击exe文件,使用默认选项安装即可。

(3)解压cudnn到当前目录,得到cuda文件夹

把这三个文件夹里的文件分别放入安装好的cuda的目录,比如:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0

路径下的对应文件夹里面,注意把lib里面x64里的文件复制过去,不要把文件夹复制过去,一个一个文件打开对应复制过去。

(4)安装tensorflow-gpu

pip install tensorflow-gpu==1.3.0 -i https://pypi.tuna.tsinghua.edu.cn/simple/

注意:Tensorflow的版本一定与安装的cuda和cudann匹配才行。

(5)测试是否安装成功

使用import tensorflow as tf,出现图中的结果即是安装成功!

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20180711G1Y7GP00?refer=cp_1026
  • 腾讯「腾讯云开发者社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 cloudcommunity@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券