鲲鹏社区首页
中文
注册
我要评分
文档获取效率
文档正确性
内容完整性
文档易理解
在线提单
论坛求助

验证ACL库功能

若开启了ACL库加速功能,需要先验证基本功能,然后验证ACL库功能,测试用例均来源于examples代码仓。

  1. 进入“/path/to/pytorch/examples/mnist”目录。
    1
    cd /path/to/pytorch/examples/mnist
    
  2. 下载Python依赖包。
    1
    pip install torchvision==0.16.2 --no-deps
    
  3. 执行以下命令运行测试用例。
    1
    python3 main.py
    

    运行结果如下:

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