win7安装tensorflow GPU版本

win7安装tensorflow1.8 GPU版本

win7下安装tensorflow,CPU版本可直接安装,无需其他配置,而GPU版本则需要安装相应程序。

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

你可能感兴趣的:(机器学习,python学习,TensorFlow)