美文网首页我爱编程
Windows安装TensorFlow-GPU流程

Windows安装TensorFlow-GPU流程

作者: 月下霜溪 | 来源:发表于2018-03-04 07:40 被阅读174次

环境:

  • Windows 10 x64
  • GTX 960

整体流程并不复杂,可分为如下五步:

  • 安装NVidia驱动
  • 安装CUDA 9.0
  • 安装cuDNN 7.x for CUDA 9.0
  • 安装Anaconda
  • pip安装tensorflow-gpu

下面逐一进行记录

一、安装NVidia驱动

这里只需要直接百度NVidia,找到NVidia官网,下载好驱动安装就可以了。其实就是在这里搜索对应的显卡驱动,进行下载。

下载完成后,重启一下电脑即可。

二、安装CUDA 9.0

本来自己之前安装的是CUDA 9.1,但是所有安装完成后,imort tensorflow时报错找不到相应的动态链接库,细看文件名发现,原来tensorflow使用的是CUDA 9.0,所以不得不重新下载安装。后来查看官方文档发现,官方文档中给出的组件正是 CUDA Toolkit 9.0,严格按照官方来,总不会错。

官网下载的时候,默认给出的是最新版9.1的下载地址,注意要选择Legacy Releases找到9.0版本,这里直接给出地址 https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

下载好 Base Installer 和 Patch 1,先安装 Base Installer,然后安装 Patch 1即可。

安装完成后,打开命令提示符,输入 echo %CUDA_PATH%,可以看到CUDA_PATH这个环境变量已经设置好了。TensorFlow官网上要求将这个变量加入到%PATH%中,加入的方式为:

搜索并打开run > 输入control sysdm.cpl > 高级 > 环境变量 > 选中 path 点击编辑 > 新建 > 输入%CUDA_PATH% > 确定保存

这样 CUDA 9.0就已经安装好了。

三、安装cuDNN v7

这里在安装的时候也是遇到坑的,就是 https://developer.nvidia.com/ 这个网站总是显示正在维护,因而也找不到下载的地方,后来在知乎上找到了解决方法:

nvidia的cudnn为什么不能下载了? - 瞎折腾的一生的回答 - 知乎

里面截图中给出的链接 https://developer.nvidia.com/rdp/cudnn-download 可以使用,进去后注册登录,同意用户协议即可下载。需要注意的是 cuDNN 有一个版本号,而同一个版本的 cuDNN 适配了不同版本的 CUDA,这需要和自己安装的CUDA版本对应起来。所以,需要下载的是 cuDNN v7.1.1 for CUDA 9.0。

安装的过程非常简单,只需要将cuda文件夹下的内容复制到%CUDA_PATH%下即可.

cuDNN/cuda %CUDA_PATH%
bin/* bin/*
include/* include/*
lib/* lib/*

四、安装Anaconda

推荐清华镜像源,下载速度有保障。这里下载的是 https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.1.0-Windows-x86_64.exe

五、安装tensorflow-gpu

1.配置pip源,使用清华镜像

https://mirrors.tuna.tsinghua.edu.cn/help/pypi/

2.安装tensorflow

开始 > Anaconda3 (64-bit) > Anaconda Prompt

输入

pip install tensorflow-gpu

相关文章

网友评论

    本文标题:Windows安装TensorFlow-GPU流程

    本文链接:https://www.haomeiwen.com/subject/ghcffftx.html