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

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