sas

最新推荐文章于 2024-01-09 11:44:46 发布

zyh88的救赎

最新推荐文章于 2024-01-09 11:44:46 发布

阅读量457

点赞数

分类专栏：杂谈

本文链接：https://blog.csdn.net/zyhzyh99/article/details/53538436

版权

杂谈专栏收录该内容

2 篇文章 0 订阅

订阅专栏

安装tensorflow前的准备工作

CUDA

CUDA(Compute Unified Device Architecture),显卡厂商NVidia(英伟达)推出的运算平台。 CUDA™是一种由NVIDIA推出的通用并行计算架构，该架构使GPU能够解决复杂的计算问题。它包含了CUDA指令集架构(ISA)以及GPU内部的并行计算引擎。
CUDA体系结构的组成：

开发库：开发库是基于CUDA技术所提供的应用开发库。
运行期环境：运行期环境提供了应用开发接口和运行期组件，包括基本数据类型的定义和各类计算、类型转换、内存管理、设备访问和执行调度等函数。
驱动：CUDA-enable的GPU的设备抽象层，提供硬件设备的抽象访问接口。也就是需要安装有nVIDIA硬件的电脑上安装相应的驱动来实现CUDA通用运算。
注意：如果之前已经安装了CUDA，安装前一定要确定版本，Ubuntu下查看CUDA版本的命令为：nvcc –V

安装tensortflow必须注意：

The GPU version works best with Cuda Toolkit 8.0 and cuDNN v5. Other versions are supported (Cuda toolkit >= 7.0 and cuDNN >= v3) only when installing from sources. Please see Cuda installation for details. For Mac OS X, please see Setup GPU for Mac.（一定要搞清楚GPU version 和Cuda toolkit版本匹配，否则后果自负！！！）
tensorflow 官网https://www.tensorflow.org/，点击get started后自动进入最新版本的页面，此时一定要小心，根据自己之前安装Cuda版本选择相应tensorflow版本，在“Download and Setup”模块部分，会介绍GPU工作最好的版本是那个版本，比如，选择tensorflow Version: r0.10，那么此时会显示The GPU version works best with Cuda Toolkit 7.5 and cuDNN v5. ，之后就按照安装向导中的步骤，一步步安装就可以了