Win10安装TensorFlow(GPU)
总述
安装到实现用了点时间(估计是自己头铁吧)。慢慢填坑慢慢摸索,终于把win10下的TensorFlow-gpu装好了,我先说遇到的最大的坑:Cuda 、Cudnn、TensorFlow这三个版本一定要相互匹配 (我在这里死了很多遍)具体的可以网上查找到(我用的是cuda和cudnn是9.0的、TensorFlow版本是1.10的、python是3.5),最重要的工具当然用的是Anaconda3。
仅作为小白入门制作,过失请不吝赐教!
准备
也就是需要下载的东西最好都从官网下载(地址自己动手找)
1、Anaconda下载并且安装(默认点击next就好,好像有一个路径环境变量那个设置可以勾选就不用手动设置环境变量了)
如下图:安装启动前若不勾选上第一项,则需在软件安装完成手动添加环境变量
2、Cuda :这个下载下载到一半容易失败或者出错(建议用百度网盘或者迅雷下载)
3、Cudnn:下载很容易,压缩包解压即可
操作
1、安装CUDA
运行cuda_9.0.176_win10.exe。安装软件会先运行一个系统检查,如果没有软硬件不兼容的情况就能继续进行下一步。若有不兼容情况,系统检查则会报错,无法进行下一步。如果报错了,就最先考虑下是不是自己的显卡不被该版本支持。
在选项着一栏选择自定义,否则安装程序会一股脑把里面所有软件包都给你装上。
在选择安装项时一般不安装GeForce Experience,CUDA是核心组件必须勾上,剩下两个选项的当前版本如果比新版本低的话也可以勾选上。接着点击下一步就开始安装了。
2、安装Cudnn
解压压缩包cudnn-9.0-windows10-x64-v7.0.zip,得到三个文件夹
将这三个文件拷贝到CUDA9.0的安装路径根文件夹下
3、安装TensorFlow
打开Anaconda Prompt进入Anaconda命令行管理界面。配置清华仓库镜,输入指令:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes
创建运行环境,输入指令:
conda create -n tensorflow-gpu python=3.5
新建一个名字叫“tensorflow-gpu”,python版本为3.5的运行环境,此环境与Anaconda中其它环境隔离。 红框中的软件包也会随之安装,输入“y“和回车后开始安装。
激活并进入环境,使后续指令在激活的环境中生效,输入指令:
conda activate tensorflow-gpu
升级pip到最新版,防止稍后的安装时,出现错误(笔者初次在安装tensorflow时,没有更新pip到最新版,导致下载到一半出现错误),输入指令:
python -m pip install --upgrade pip
验证
激动人心的时刻到来了,对于DL的小白来说,把ubantu(cpu版略)和window(GPU)下的TensorFlow装好,并成功验证,真的是一件不错的开心事,愿各位从此在DL界留名青石。
在命令行中,进入python,并输入以下代码:
import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))
输出:
b’ Hello, TensorFlow!
恭喜你,进入DL学习!