Windows 10安装TensorFlow之CUDA篇
TensorFlow使用GPU版本效率会更高,本文将讲解TensorFlow-GPU版本的安装,由TensorFlow官网可知安装GPU版本时所需要的硬件和软件的requirements,它需要NVIDIA的GPU(硬件)和NVIDIA 驱动、CUDA、CUDNN(软件),其他可以不用。
接下来将讲解CUDA的安装。
版本对应
前提:你有支持CUDA的NVIDIA GPU
-
查看TensorFlow版本对应:
TensorFlow版本 Python 版本 cuDNN CUDA TensorFlow >= 1.13.0 3.5-3.6 cuDNN >= 7.4.1 CUDA 10.0 <注:国外的需要梯子,2019/10/12查看和国内一样的>
-
安装最新版本的驱动(会很重要),官网下载即可。
所有NVIDIA显卡版本的驱动下载地址:驱动下载
NVIDIA GeForce 显卡去驱动下载地址:驱动下载
-
查看驱动中的系统信息,找支持的CUDA的版本
a. 桌面(右键)–> NVIDIA 控制面板–> 帮助–> 系统信息–> 显示–> 在细节中查看驱动程序版本
b. 到NVIDIA官网查看驱动程序对应的CUDA的版本–> CUDA Driver
-
查看TensorFlow-GPU版本要求
a. 去TensorFlow官网查看GPU版本要求–> Software requirements注:此时可以匹配你要的TensorFlow版本了一般版本1到版本2在NVIDIA大部分显卡上都可以运行了
由此就知道你的TnsorFlow、CUDA、cuDNN的版本了
安装
按照前面对应的版本去下载即可