tensorflow-gpu 配置安装【精简备忘】

首先确定GPU支持：

如果装了cpu版: pip uninstall tensorflow

依赖官方解释：https://tensorflow.google.cn/install/gpu

 

主要安装三个东西：tensorflow-gpu、CUDA、cuDNN 和 %环境变量%

1、pip install tensorflow-gpu

2、目前官方支持9.0（10.0好像也可以），CUDA：https://developer.nvidia.com/cuda-90-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal

路径可改，其他默认下一步，安装后打开安装目录……/CUDA Samples/v9.0，等会将cuDNN解压到此目录下

 

 

3、cuDNN:https://developer.nvidia.com/rdp/cudnn-download

选相应版本及相应系统

解压到……/CUDA Samples/v9.0/

 

4、配置环境变量（3个）：

……/CUDA Samples/v9.0/bin

……/CUDA Samples/v9.0/cuda/bin

……/CUDA Samples/v9.0/cuda/lib/x64

 

至此 成功import tensorflow即表示配置成功，运行session会出现gpu字样。