sas

来源:互联网 发布:中银淘宝校园卡 编辑:程序博客网 时间:2024/06/09 15:13

安装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. ,之后就按照安装向导中的步骤,一步步安装就可以了

0 0
原创粉丝点击