Windows环境下Anaconda安装TensorFlow的避坑指南

最近群里聊天时常常会提到DL的东西,也有群友在学习mxnet,但据说坑比较多。为了遇上潮流顺便避坑,我果断选择了TensorFlow,然而谁知一上来就掉坑里了……python

我根据网上的安装教程,默认安装了最新版本的CUDA和cuDNN,但以后遇到了各类没法导入的问题,后来我发现主要的问题就是版本问题!!!CUDA版本和TensorFlow版本不匹配是最重要的坑。windows

目前NV家最新的组件是CUDA 9.1+cuDNN 7.05,而咱们安装所用的清华大学的镜像里TensorFlow是1.3.0这两个是无法匹配的,正确的配置是CUDA 8.0+cuDNN 6.0+TensorFlow 1.3.0,另外还有python 3.5网络

 

接下来讲正事,Anaconda目前的python版本是3.6的,但不要紧,咱们新建一个环境就能够了。首先更新一下N卡驱动,并下载并安装这几样东西:app

VisualC++ Redistributable for Visual Studio 2015学习

CUDA Toolkit 8.0 (注意操做系统和版本的选择)url

cuDNN 6.0 (注意,要选Download cuDNN v6.0 (April 27, 2017), for CUDA 8.0spa

下载完成后逐一安装,CUDA安装时候可能会提示警告,主要说你的显卡驱动可能太新或者VS版本有点问题什么的,不过只要你以前的步骤都妥了就不用理它。操作系统

cnDNN解压缩,里面会有3个文件夹:命令行

把这三个复制到CUDA的安装目录,好比C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0,与同名文件夹合并便可。以后就是安装TensorFlow了。3d

 

这里基本按照一位群友上传的安装教程来,打开CMD,以此输入以下命令:

一、更改Anaconda镜像源

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/ (执行两次保证将该镜像放在首位)
conda config --set show_channel_urls yes

二、利用Anaconda建立一个python3.5的环境,环境名称为tensorflow :

conda create -n tensorflow python=3.5

三、激活该环境:

activate tensorflow

四、安装CPU版本和GPU版本的TensorFlow:

pip install --upgrade --ignore-installed https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/cpu/tensorflow-1.3.0rc0-cp35-cp35m-win_amd64.whl
pip install --upgrade --ignore-installed https://mirrors.tuna.tsinghua.edu.cn/tensorflow/windows/gpu/tensorflow_gpu-1.3.0rc0-cp35-cp35m-win_amd64.whl

这一步中有时候也会报错,只要你看到过进度条,那就没啥大问题,主要是网络很差的问题……若是网络很差报错了,就再执行一次一样的命令,最终必定会成功。

五、安装完成后,记得设置环境变量,环境变量path里加入如下内容:

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin; (你CUDA的安装位置,默认的)
C:\Program Files\NVIDIA Corporation\NVSMIs;
D:\Anaconda3\envs\tensorflow\Scripts; (这两个是你Anaconda的安装位置,我装在D盘里,后面的\envs\tensorflow是指你在第2步中建立的tensorflow环境)
D:\Anaconda3\envs\tensorflow\Library\bin;

六、试一试可否成功

打开cmd,输入activate tensorflow激活tensorflow环境,输入python进入python,尝试输入一下命令

import tensorflow as tf
hello = tf.constant('Hello, TensorFlow!')
sess = tf.Session()
print(sess.run(hello))

成功的话你会看到以下画面

七、配置spyder

Anaconda最吸引个人就是spyder,集成了命令行和IDLE,超好用,然而安装完咱们会发现一个问题,你打开spyder是没法导入tensorflow的,提示找不到模块。这主要是由于咱们打开的spyder是root环境下的,里面并无tensorflow。因此咱们要在咱们以前建立的tensorflow环境中安装spyder才行。打开Anaconda Navigator能够看到tensorflow环境下只有这些库,咱们能够根据本身的须要再安装一下其余须要的库,包括spyder

在库上方选Not installed(未安装的库),而后在右侧搜索spyder,选中以后点最右下角的apply就会安装了。

安装完成后打开开始菜单,咱们会看到多出来一个spyder(tensorflow),就是这个啦~

 能够看到如今也能够用tensorflow了~

若是你点击的时候只是光标闪了一下并无打开spyder的话,直接在Anaconda Navigator的home界面中点击spyder的Launch,就能启动了

启动后咱们再试试导入,也没问题,大功告成啦~

相关文章
相关标签/搜索