AMCT (TensorFlow, Ascend)

This scenario applies only to Atlas training product.

Install software as the AMCT installation user. If the said user is a non-root user, check that the user has the sudo permission. Then run the su - username command to switch to the non-root user and run the commands listed in the following table. (If the installation user is root, delete --user from the dependency installation command.)

Install dependencies by referring to "Appendix A: Reference > Dependencies" in CANN Software Installation. After the installation is complete, run the following commands to install the dependencies (the Pillow software is required for image processing):

Table 1 Dependencies

Dependency

Version

Installation Command

CPU version of TensorFlow

2.6.5 or 1.15.0

Select one based on your requirements.

  • CPU version of TensorFlow 2.6.5
    python3 -m pip install tensorflow-cpu==2.6.5 --user
  • CPU version of TensorFlow 1.15
    python3 -m pip install tensorflow-cpu==1.15 --user
NOTICE:
  • If different TensorFlow versions are used, ensure that the Python version in your installation command is correct. (Change the environment variables to match the Python version in use.)
  • Currently, quantization is supported only in TensorFlow 1.15.0, whereas inferences during quantization can be run in both TensorFlow 1.15.0 and 2.6.5.

Pillow

6.0.0 or later

pip3 install pillow==6.0.0 --user