验证ACL库功能
若开启了ACL库加速功能,需要先验证基本功能,然后验证ACL库功能,测试用例均来源于examples代码仓。
- 进入“/path/to/pytorch/examples/mnist”目录。
1
cd /path/to/pytorch/examples/mnist
- 下载Python依赖包。
1
pip install torchvision==0.16.2 --no-deps
- 执行以下命令运行测试用例。
1
python3 main.py
运行结果如下:
mnist用例包含多个Epoch,训练时间较长,验证功能时执行完一个Epoch的训练即可。
父主题: 运行和验证