TensorFlow安装

Windows

在CUDA与CUDNN安装好之后,查看版本

  1. 获取CUDA版本: 控制面板->设定小图标->Nvidia控制面板->组件标签栏->CUDADLL查看版本(我这里9.2.217)
  2. 获取CUDNN版本: 打开默认路径 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0includecudnn.h(我这里7.0.5)

然后安装GPU版本的TensorFlow(如果已经安装这个操作会升级到最新版本)

pip3 install --upgrade tensorflow-gpu

原文地址:https://www.cnblogs.com/xytpai/p/13038673.html