Windows10 下的 Tensorflow 环境配置

前言

正文

安装 tensorflow

通过 pip 安装

开启 GPU 支持

找到对应版本的CUDA、cuDNN

查看官方的页面 GPU 支持  |  TensorFlow 可以得到 CUDA 和 cuDNN 的版本

下载 CUDA

安装 CUDA

下载 cuDNN

放置 cuDNN

配置环境变量

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\libnvvp

参考

GPU 支持  |  TensorFlow
Windows TensorFlow安装教程（gpu和cpu）
Windows10+Cuda10.0+cuDNN+Anaconda+Tensorflow-gpu+Keras安装教程