因为墙,下载tensorflow-gpu太慢,所以写下这篇东西。
网上具体教程太多,看得头晕还不知是否合适自己,少见有提供思路的。
因为程序总是在不断更新,所以每个步骤具体怎么做→百度,此处给小白提供思路,减少爬坑时间。
显卡决定驱动程序,驱动程序决定cuda、cudnn版本,cuda、cudnn版本决定tensorflow-gpu版本,或者能不能用tensorflow-gpu。
查看自己的显卡型号是不是支持cuda(tensorflow-gpu需要这东西)
https://developer.nvidia.com/cuda-gpus
cuda和显卡驱动的对应关系
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
查看tensorflow-gpu和cuda、cudnn版本关系,决定安装的tensorflow-gpu、cuda、cudnn版本
https://tensorflow.google.cn/install/source
下载安装对应显卡驱动
windoes:
https://www.nvidia.cn/Download/index.aspx?lang=cn
ubuntu:
用自带的:软件和更新-附带驱动,下载安装
下载对应cuda
https://developer.nvidia.com/cuda-toolkit-archive
下载对应cudnn(需注册)
https://developer.nvidia.com/rdp/cudnn-download
接下来:
安装cuda、cudnn
安装对应版本tensorflow-gpu
测试tensorflow-gpu
完。
博主也是门外汉,有错的或者需要改正的,请提示。