曦远Code

摘要： 前言 在上一篇文章中，我们已经在 Windows + AMD显卡 + ROCm 环境下，成功用 PyTorch 训练了一个简单的 CNN 模型（详情可戳：训练MNIST数字识别模型）。不过那个任务相对基础，甚至普通CPU都能快速跑完，很难发挥出AMD显卡的性能优势。 所以今天，我们来挑战一些真正需要 阅读全文