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

验证BLAS库功能

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

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

    运行结果如下:

    模型训练时间较长,功能验证不需要等待模型训练完,几个迭代后手动停止即可。