TensorFlow GPU版本Windows下安装_win tensorflow gpu-优快云博客

本文链接：https://blog.youkuaiyun.com/soledad_chao/article/details/104852141

本文详细介绍如何安装CUDA10.0和cuDNN7.6.5，包括从官方网站下载相应版本，设置环境变量，以及使用pip安装TensorFlow-gpu 2.0.0。文章还提供了验证安装是否成功的Python代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

cuda 10.0和cudnn 7.6.5
注意：此处选择版本，应根据官网对GPU的测试来选择，网址为：https://tensorflow.google.cn/install/source_windows

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

https://developer.nvidia.com/rdp/cudnn-archive （注意对应cuda版本）

安装cuda 10.0，解压cudnn，将文件粘贴到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0）
添加环境变量到PATH

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\include

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64 

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\CUPTI\libx64

pip安装TensorFlow：
pip insatall tensorflow-gpu==2.0.0
注意：此处未安装2.1.0是因为没有对应的cuda。
运行代码：

from tensorflow.python.client import device_lib
import os

os.environ["TF_CPP_MIN_LOG_LEVEL"] = "99"
print(device_lib.list_local_devices())

出现以下代码即为成功：

[name: "/device:CPU:0"
device_type: "CPU"
memory_limit: 268435456
locality {
}
incarnation: 15018807026964148199
, name: "/device:GPU:0"
device_type: "GPU"
memory_limit: 3149044121
locality {
  bus_id: 1
  links {
  }
}
incarnation: 14256059926015496502
physical_device_desc: "device: 0, name: GeForce GTX 1050, pci bus id: 0000:01:00.0, compute capability: 6.1"
]