tensorflow -gpu配置
在安装tensorflow -gpu时候,python, cudnn,CUDA以及NVIDIA驱动的版本都要一一对应。搞了好久,才将版本一一匹配。展示一下我安装的对应版本
基于win10
- python 3.6
- Tensorflow-gpu 2.0.0-alpha,下载链接 tensorflow gpu-2.0.0a0-cp3.6-cp36m-win_amd64.whl
- CUDA 10.0.13 不能是10.1
CUDA 10.0.13 - cudnn 版本号大于7.4.1(推荐7.6.0),官网下载需要注册,我已经下载好了存在百度云中
cudnn百度云下载https://pan.baidu.com/s/1RhyXK8Y7IUvpA_jc4hRCxw
提取码:ub91 - NVIDIA 驱动程序410.x或更高 (下载显卡对应的驱动,或者直接在电脑中检查更新到最新版本就🆗了)