Hello TensorFlow 二 (GPU)

时间:2022-08-31 14:35:56

官方说明:https://www.tensorflow.org/install/

环境:

操作系统 :Windows 10 家庭中文版

处理器 : Intel(R) Core(TM) i7-7700 CPU @3.6GHZ 3.60GHZ

内存 :16GB

显卡:NVIDIA GeForce GTX 1060 6GB

Python:3.6.2

安装GPU版

pip3 install --upgrade tensorflow-gpu

Hello TensorFlow 二    (GPU)

安装 NVIDIA CUDA® Toolkit 8.0.

下载地址

https://developer.nvidia.com/cuda-downloads

Hello TensorFlow 二    (GPU)

安装 cuDNN v6

下载地址:https://developer.nvidia.com/rdp/cudnn-download

需要注册用户

Hello TensorFlow 二    (GPU)

将下载后的zip文件解压到c盘 C:\cudnn-8.0-windows10-x64-v6.0

将 C:\cudnn-8.0-windows10-x64-v6.0\cuda\bin 加入到环境变量Path

验证tensorflow-gpu安装

Hello TensorFlow 二    (GPU)

执行sess = tf.Session()

后,会显示显卡GPU信息。