最近群里聊天时常常会提到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.0)spa
下载完成后逐一安装,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,就能启动了
启动后咱们再试试导入,也没问题,大功告成啦~