tensorflow gpu 安装(windows 10)

1. 安装Nvidia cuda 10.0

https://developer.nvidia.com/cuda-toolkit-archive

2. 安装cudnn(选择对应的版本)

https://developer.nvidia.com/rdp/cudnn-download

tensorflow gpu 安装(windows 10)

3. 解压cudnn将里面的文件夹复制到cuda的安装目录

4. pip3 install --ignore-installed --upgrade tensorflow-gpu --user 

--user :解决permission

--ignore-installed : 如果已经安装cpu版

5.增加环境变量

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

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

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

即安装目录下bin, extras, include

6.验证

tensorflow gpu 安装(windows 10)