Windows 10安装TensorFlow之CUDA篇
Windows 10安装TensorFlow之CUDA篇TensorFlow使用GPU版本效率会更高,本文将讲解TensorFlow-GPU版本的安装,由TensorFlow官网可知安装GPU版本时所需要的硬件和软件的requirements,它需要NVIDIA的GPU(硬件)和NVIDIA 驱动、CUDA、CUDNN(软件),其他可以不用。接下来将讲解CUDA的安装。版本对应前提:你有支...
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的版本了
安装
按照前面对应的版本去下载即可

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。
更多推荐
所有评论(0)