Windows 10 +Anaconda+tensorflow+cuda8.0 环境配置

来源:互联网 发布:魅族手机源码 编辑:程序博客网 时间:2024/06/18 03:43

tensorflow要求python3.5版本,anaconda2(python2.7)和3(python3.6)都需要额外安装3.5。因为已经装了anaconda2,本教程就用2好啦。

1. 安装Anaconda

官网地址:
https://www.continuum.io/downloads

下载完毕后打开安装包,一步一步安装就可以了。

注意安装之后需要把相关路径加入环境变量:

D:\Anaconda2D:\Anaconda2\ScriptsD:\Anaconda2\Library\bin

2. CUDA 8.0

1. 下载cuda8.0:

官网地址:https://developer.nvidia.com/cuda-downloads
百度云:下载cuda_8.0.61_win10.exe

2. 安装cuda8.0:

双击cuda_8.0.61_win10.exe直接进行安装即可,默认安装到C:\ProgramFiles\NVIDIA GPU Computing Toolkit目录下;

3. 验证cuda8.0已正确安装:

在cmd命令行输入:

nvcc -V

这里写图片描述

安装完之后系统变量会自动添加上。

3. cuDNN 5.1

官网下载(需登录) :https://developer.nvidia.com/rdp/cudnn-download
百度云:下载cudnn-8.0-windows10-x64-v5.1.zip

解压后分别将三个文件放到cuda的相应安装目录下:

C:\cuda\bin\cudnn64_5.dll —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\cuda\include\cudnn.h —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
C:\cuda\lib\x64\cudnn.lib —> C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib\x64

4. Tensorflow

TensorFlow目前在Windows下只支持Python 3.5版本。所以需要用conda create创建一个python3.5的环境。

1. 打开Anaconda Prompt,输入清华仓库镜像,这样更新会快一些:

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

2. 同样在Anaconda Prompt中利用Anaconda创建一个python3.5的环境,环境名称为tensorflow ,输入下面命令:

conda create -n tensorflow python=3.5
  • -n tensorflow: 环境的名字为tensorflow
  • 打开环境:‘activate tensorflow’
  • 关闭环境:“deactivate tensorflow‘’

3. 在Anaconda Prompt中启动tensorflow环境:

activate tensorflow

这里写图片描述

4. 安装gpu版本的TensorFlow

pip install --ignore-installed --upgrade https://storage.googleapis.com/tensorflow/windows/gpu/tensorflow_gpu-1.1.0-cp35-cp35m-win_amd64.whl 

5. 测试tensorflow

打开anaconda prompt,激活环境并进入python:

这里写图片描述

测试代码如下:

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

运行结果:

这里写图片描述

参考:

  1. http://blog.csdn.net/u010858605/article/details/64128466
  2. http://www.linuxidc.com/Linux/2016-12/138862.htm
  3. http://blog.csdn.net/u010858605/article/details/64128466
阅读全文
0 0