文档HOT
下载
中文
注册

AMCT（TensorFlow）

请使用AMCT的安装用户安装依赖的软件，如果安装用户为非root，请确保该用户拥有sudo权限，请使用su - username命令切换到非root用户执行如下命令。

表1 依赖列表

依赖名称

版本号

安装命令

Python

3.9.2或3.7.5

Ubuntu、KylinV10SP1、openEuler和UOS操作系统请参见安装Python3.9.2（Ubuntu）。

TensorFlow

2.6.5或1.15.0

用户根据实际情况选择安装CPU或GPU。

  • TensorFlow 2.6.5 CPU版本安装命令：
    python3 -m pip install tensorflow-cpu==2.6.5 --user
  • TensorFlow 2.6.5 GPU版本安装命令：
    python3 -m pip install tensorflow-gpu==2.6.5 --user
  • TensorFlow 1.15 CPU版本安装命令：
    python3 -m pip install tensorflow-cpu==1.15.0 --user
  • TensorFlow 1.15 GPU版本安装命令：
    python3 -m pip install tensorflow-gpu==1.15.0 --user
须知：
  • 使用不同的TensorFlow版本时，请确保上述命令中的Python版本对应正确（切换所使用Python版本的环境变量即可）。
  • 当前仅支持TensorFlow 1.15.0版本的网络模型进行量化，但量化过程中支持在TensorFlow 1.15.0、2.6.5版本环境进行推理。

numpy

1.16.0~1.23.5或1.20.0~1.23.5

 
pip3 install numpy==1.16.0 --user
或pip3 install numpy==1.20.0 --user

protobuf

3.13.0+

 
pip3 install protobuf==3.13.0 --user
父主题： 安装依赖