验证ACL库功能
若开启了ACL库加速功能,需要先验证基本功能,然后验证ACL库功能,测试用例均来源于examples代码仓。
- 进入“/path/to/pytorch/examples/mnist”目录。
1cd /path/to/pytorch/examples/mnist
- 下载Python依赖包。
1pip install torchvision==0.16.2 --no-deps
- 执行以下命令运行测试用例。
1python3 main.py运行结果如下:

mnist用例包含多个Epoch,训练时间较长,验证功能时执行完一个Epoch的训练即可。
父主题: 运行和验证