Windows 7下安装TensorFlow1.6(cuda9.0+cuDNN7.0+python3.5+pip9)
Windows 7下安装TensorFlow1.6(cuda9.0+cuDNN7.0+python3.5+pip9)
1 选择要安装的版本
有两类TensorFlow可供选择
1.1 TensorFlowwithCPU support only
适用于没有NVIDIA® GPU的情况,安装更加简单快捷。
1.2 TensorFlowwithGPU support(选择这个)
适用于有NVIDIA® GPU的情况,Tensor程序在GPU上比在CPU上运行效率更高。
2 准备工作(安装带有GPU支持的TensorFlow)
2.1 安装CUDA
2.1.1 确认你的GPU支持CUDA
我的电脑的GPU是GeForce GTX 750 Ti,支持CUDA (compute capability: 5.0))。
基本主流的英伟达显卡都支持CUDA,欲查看支持CUDA的所有GPU的列表可以访问:
http://developer.nvidia.com/cuda-gpus
2.1.2 下载并安装NVIDIA CUDAToolkit
下载地址为:
https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=7&target_type=exelocal
2.2 安装cuDNN
2.2.1 下载cuDNN
cuDNN的下载地址如下:(需要注册)
https://developer.nvidia.com/cudnn
请注意cuDNN的版本一定要和CUDA相适应。
近期Nvidia维护cuDNN下载页,需要从第三方下载:
https://www.7down.com/soft/236025.html
2.2.2 解压cuDNN
将下载到的安装包解压到:
D:\ProgramFiles\cuDNN\
我的CUDA的安装路径是:
C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.0
2.2.3 添加头文件和库文件到CUDA
复制D:\ProgramFiles\cuDNN\目录下的三个文件夹,粘贴到C:\ProgramFiles\NVIDIA GPU ComputingToolkit\CUDA\v9.0路径下(会与原有的同名文件夹合并)。
2.2.4 添加环境变量
检查环境变量名:CUDA_PATH值:C:\ProgramFiles\NVIDIA GPU Computing Toolkit\CUDA\v9.0是否存在,不存在则添加。
2.2.5 在visual studio中开发(省略)
应当包含库文件cudnn.lib
a) 右键单击工程名
b) 依次点击 Linker > Input>Additional Dependencies
c) 添加cudnn.lib并确认即可。
3 决定以何种方式安装TensorFlow
可选的安装方式有两种: (方式2比较繁琐,选1)
1. "native"pip
2. Anaconda
3.1 以native pip方式安装
3.1.1 安装python3(选Python3.5,Python3.6测试没有通过)
TensorFlow在Windows上仅支持Python 3.5.x和3.6.x。
安装Python3.5需要升级pip
python -m pip install --upgrade pip
否则会提示:
You are using pipversion 8.1.1, however version 9.0.1 is available.
You should consider upgrading via the 'python -m pip install --upgrade pip'comm
and.
3.1.2 安装TensorFlow(没有测试)
安装TensorFlow的命令如下:
安装仅支持CPU的TensorFlow
pip3install--upgradetensorflow
安装支持GPU的TensorFlow
pip3install--upgradetensorflow-gpu
3.2 以Anaconda的方式安装(很麻烦,不考虑)
3.2.1 下载并安装Anaconda
下载地址如下:
https://www.anaconda.com/download/
图形界面一路安装即可。
3.2.2 创建一个conda环境
conda create -n tensorflow python=3.5
3.2.3 **conda环境
activate tensorflow
3.2.4 安装TensorFlow
安装仅支持CPU的TensorFlow
pipinstall--ignore-installed--upgradetensorflow
安装支持GPU的TensorFlow
pipinstall--ignore-installed--upgradetensorflow-gpu
4 验证安装是否成功
打开python控制台,输入如下代码:(第三条执行时间较长)
>>> import tensorflow as tf
>>> hello = tf.constant('Hello, TensorFlow!')
>>> sess = tf.Session()
>>> print(sess.run(hello))
若显示如下:
b‘Hello, TensorFlow!
则说明安装成功。
https://blog.****.net/duoker/article/details/79483434