Windows
在CUDA与CUDNN安装好之后,查看版本
- 获取CUDA版本: 控制面板->设定小图标->Nvidia控制面板->组件标签栏->CUDADLL查看版本(我这里9.2.217)
- 获取CUDNN版本: 打开默认路径 C:Program FilesNVIDIA GPU Computing ToolkitCUDAv9.0includecudnn.h(我这里7.0.5)
然后安装GPU版本的TensorFlow(如果已经安装这个操作会升级到最新版本)
pip3 install --upgrade tensorflow-gpu