文档HOT
下载
中文
注册

AMCT（TensorFlow,Ascend）

该场景仅限Atlas 训练系列产品使用。

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

请参见《CANN 软件安装指南》>“附录：参考信息 > 依赖列表”章节，安装完成后，还需要执行如下命令安装相关依赖，对图像处理时需要依赖Pillow软件：

表1 依赖列表

依赖名称

版本号

安装命令

TensorFlow CPU版本

2.6.5或1.15.0

用户根据实际情况选择如下版本进行安装：

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

Pillow

6.0.0+

 
pip3 install pillow==6.0.0 --user
父主题： 安装依赖