Rate This Document
Findability
Accuracy
Completeness
Readability

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.

  1. Go to the /path/to/pytorch/examples/mnist_forward_forward directory.
  2. Download the Python dependency package.
    pip install torchvision==0.16.2 --no-deps
  3. 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.