TensorFlow安装

Windows

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

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

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

pip3 install --upgrade tensorflow-gpu

posted @ 2020-06-03 17:11  xytpai  阅读(110)  评论(0)  编辑  收藏  举报