腾讯云提供的云GPU服务器性能强大, 费用合理, 所以笔者试用腾讯云GPU云服务器完成了ViT模型的离线训练, 并记录了试用过程, 以供参考。...下面具体介绍使用腾讯云GPU服务器训练ViT模型的过程。
GPU云服务器初始化
首先我们前往腾讯云GPU选购页面进行选型。...目前提供了GN7与GN8两种规格的服务器可选:
[云服务器规格]
根据Technical提供的GPU对比, Turing架构的T4性能优于Pascal架构的P40, 所以优先选用GN7。...通过腾讯云在线Ping工具, 从笔者所在位置到提供GN7的重庆区域延迟最小, 因此选择重庆区域创建服务器。...(我们也可以通过增加模型的参数量, 如修改模型为vit\_small\_patch16\_224, 来进一步尝试优化模型效果):
[训练结果]
总结
本文记录了试用腾讯云GPU服务器训练一个ViT图像分类模型的过程