首先说明下:GPU驱动有很多匹配方案,匹配方案可以是tensorflow 1.5+cuda 9+cudnn7或者tensorflow1.4+cuda 8+cudnn6或者tensorflow1.2+cuda 8+cudnn5,安装时从这些方案选一个即可,这里举例tensorflow1.2+cuda 8+cudnn5方案。
1.下载驱动
(1)下载cuda 8
地址为https://developer.nvidia.com/cuda-80-ga2-download-archive,
选项截图为:
从中下载得到cuda 8,有时候一会就直接完成了,包的大小不对,那是下载失败了。
(2)下载cudnn 5
地址为https://developer.nvidia.com/rdp/cudnn-archive
选项截图为:
需要登录账号下载,