在 cuda 架构上启用 Tensor 核心的使用

请确保在 cuda 架构上能够利用 Nvidia 张量核心,这样 机器学习 项目就可以集成到 taichi 中。这也有助于加快其他类型的计算。您应该清楚,这正是 cuda 比 taichi 更具优势的地方。在矩阵乘法方面,20 倍的性能是不可小觑的。 祝你好运,继续努力。