我要评分
获取效率
正确性
完整性
易理解

Verifying BLAS

If BLAS acceleration is enabled, verify its basic functions and then BLAS. 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.
    1
    pip install torchvision==0.16.2 --no-deps
    
  3. Run the test case.
    1
    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.