Verifying BLAS
If BLAS acceleration is enabled, verify its basic functions and then the BLAS function. The test cases are all from the examples code repository.
- Go to the /path/to/pytorch/examples/mnist_forward_forward directory.
- Download the Python dependency package.
pip install torchvision==0.16.2 --no-deps
- Run the test case.
python3 main.py
The execution result is as follows:

The model training takes a long time. You do not need to wait until the model training is complete. You can manually stop the process after several iterations.
Parent topic: Running and Verification