win7安装tensorflow1.8 GPU版本
win7下安装tensorflow,CPU版本可直接安装,无需其他配置,而GPU版本则需要安装相应程序。
- 安装vs2015以上的版本,windows官网下载社区版即可。
- 安装tensorflow需要的CUDA版本,例如GPU1.8的需要CUDA9,去NVIDIA官网下载 https://developer.nvidia.com/cuda-90-download-archive
- 下载匹配TensorFlow版本的cuDNN,比如目前TensorFlow-gpu 1.8需要cuDNN v7.0.版https://developer.nvidia.com/rdp/cudnn-archive
- 添加环境变量
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\lib
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0\include
将四个环境变量添加,一般是后两个就可以 - cuDNN压缩包解压后得到3个子文件夹(bin,lib,include),你需要把这3个子文件夹复制“NVIDIA GPU Computing Toolkit\CUDA\v9.0”文件夹下(系统会提示已经存在,是否合并,选择确定就可以了)
- 打开CMD使用anaconda中的pip命令安装tensorflow
pip install tensorflow-gpu==1.8 -i https://pypi.tuna.tsinghua.edu.cn/simple - 安装完成后,运行Python,使用import tensorflow as tf 进行测试