PyTorch MNIST 官方示例代码在 CNB 中的运行实现。

1. 运行以下命令可持续监测显卡利用情况。

   watch -n 1 nvidia-smi
   

2. 运行 check-cuda.py 可检测显卡的 CUDA 支持情况。

3. 运行 mnist.py 可开始训练和测试 MNIST 数据集。