Environment Setup
- Install Driver, Firmware, CANN Toolkit, and ops operator package, and set up the development environment and operating environment. For details, see CANN Software Installation Guide.
- Install PyTorch training software by referring to the Ascend Extension for PyTorch Software Installation Guide.
- Install third-party dependencies.
After the environment is set up, install dependent third-party software based on Table 1.
Table 1 AOE dependent software Third-party Software
Description
How to Install
pciutils
For AOE tuning, the lspci command is used to obtain the hardware device where the Ascend AI Processor is located. Therefore, the pciutils tool must be installed in the environment where the AOE tool is executed.
Example:
- Ubuntu: sudo apt-get install pciutils
openEuler: sudo yum install pciutils
For other operating systems, replace the installation command as required.
- Ubuntu: sudo apt-get install pciutils
Parent topic: Offline Tuning in PyTorch-based Training Scenarios