1.安装3.5以上Python
相关教程太多啦,这里就略了。
2.安装numpy
在命令行中执行 pip3 install numpy 即可。
3.使用pip安装TensorFlow
在命令行中执行 pip3 install --upgrade tensorflow-gpu 即可(请确保你即安装了vc++2015运行库)
4.确认你使用的是N卡并支持CUDA,从英伟达官网下载CUDA9.0并安装(目前WIN版TensorFlow只支持到CUDA9.0)
地址:https://developer.nvidia.com/cuda-90-download-archive
下载本地安装包(local),安装过程中建议选择自定义模式,并更改(或记下)安装路径,以免安装之后找不到CUDA在哪里。
5.安装cuDNN
地址:https://developer.nvidia.com/cudnn
需要注册后下载,解压后将目录中的文件复制到CUDA安装位置的相应目录下。
6.配置环境变量
在 控制面板\系统和安全\系统\高级系统设置 的高级选项卡中点击环境变量,将CUDA安装位置如xxx/CUDA/v9.0/bin添加到系统环境变量PATH中。
7.测试
在python命令行中输入 import tensorflow 并执行,若无报错则安装成功。
8.Enjoy!