下载
中文
注册

AMCT(TensorFlow)

本章节以Ubuntu 20.04为例,详述安装前准备,版本配套信息如下:

表1 配套版本信息

类别

版本限制

获取方式

备注

操作系统及版本

EulerOS release 2.0 (SP10) aarch64

cat /etc/*release && uname -m

仅支持基于CPU量化。

操作系统及版本

Ubuntu 20.04 x86_64

请从Ubuntu官网下载对应版本软件进行安装,安装完成后查询命令为:

cat /etc/*release && uname -m

支持基于CPU,GPU的量化。

操作系统及版本

Ubuntu 20.04 aarch64

请从Ubuntu官网下载对应版本软件进行安装,安装完成后查询命令为:

cat /etc/*release && uname -m

支持基于CPU,GPU的量化。

操作系统及版本

KylinV10SP1 x86_64

-

仅支持基于CPU量化。

操作系统及版本

openEuler 20.03 x86_64

-

仅支持基于CPU量化。

操作系统及版本

UOS20 aarch64

-

仅支持基于CPU量化。

CUDA toolkit/CUDA driver

11.2或10.0

请用户自行获取相关软件包进行安装,例如可以参见如下链接获取相关toolkit包,该包中包括driver软件包。

https://developer.nvidia.com/cuda-toolkit-archive

如果使用GPU模式执行量化功能,则CUDA软件必须安装。

  • TensorFlow2.6.5版本对应的CUDA版本为11.2
  • TensorFlow1.15.0版本对应的CUDA版本为10.0。

Python

Python3.7.x、Python3.8.x、Python3.9.x、Python3.10.x

推荐使用Python3.9.2

Ubuntu、KylinV10SP1、openEuler和UOS操作系统请参见安装Python3.9.2(Ubuntu)

EulerOS操作系统请参见安装Python3.9.2(EulerOS )

  • 安装依赖时,请确保服务器能够连接网络。
  • 本手册以Python3.9.2为例进行介绍,相应环境变量和安装命令以实际安装Python版本为准。

TensorFlow

2.6.5或1.15.0

请参见安装依赖

TensorFlow版本与Python版本的对应关系:

  • 2.6.5版本配套的Python版本为:Python3.7.x(3.7.5~3.7.11)、Python3.8.x、Python3.9.x
  • 1.15.0版本配套的Python版本为:Python3.7.x(3.7.5~3.7.11)

numpy

1.16.0~1.23.5或1.20.0~1.23.5

请参见安装依赖

  • 若安装的TensorFlow版本为2.6.5,则numpy安装版本必须为1.20.0~1.23.5。
  • 若安装的TensorFlow版本为1.15.0,则numpy安装版本为1.16.0~1.23.5。

protobuf

3.13.0+

请参见安装依赖

-