Windows10 下的 Tensorflow 环境配置

本文最后更新于:2020年11月22日 晚上

前言

正文

安装 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 安装教程