很多做模型训练的团队,都会先遇到本地显卡不够用、采购周期又长的问题。这时候租一台云上GPU服务器通常更灵活。华为云在北京有成熟的机房资源,离国内用户近,网络延迟低,适合国内业务部署。这篇文章按实际操作的顺序,讲北京华为云GPU服务器搭建从买机器到跑通PyTorch的过程,尽量帮你少踩几个坑。

一、为什么优先选北京区域
北京区域的好处主要是离用户近。团队如果要频繁上传数据集、调试接口,数据回传快不快,体感差别会很明显。另外,政务、金融、教育这类项目对数据存放位置有要求,选境内节点在合规审查时更好说明。具体要不要满足,还是要看项目自己的合规条款,别只凭印象定。
二、购买前先想清楚四件事
- 算力需求:先估算显存大小和训练时长。单卡推理和多卡训练对规格的要求差很多。规格选大了是浪费,选小了训练跑到一半会报显存不足。
- 计费方式:短期实验用按需计费更划算,试错成本低。如果确定要长期稳定运行,再考虑包年包月。
- 网络规划:提前定好VPC网段、子网和公网带宽。后期改网段要迁移,麻烦不小。
- SSH密钥:创建实例时直接绑定密钥,比密码登录更安全,也省得之后再改配置。
三、创建GPU实例的步骤
- 登录华为云控制台,地域选北京,进入弹性云服务器页面。创建前确认一下地域,建错了之后资源不好迁移。
- 在规格里筛选GPU加速型,按显卡型号和显存选实例。
- 镜像可以选公共镜像里的Ubuntu或CentOS。如果平台有现成的AI镜像,驱动和部分依赖已经装好,能省不少时间。是否包含你需要的CUDA版本,要在镜像说明里确认。
- 网络选已经规划好的VPC,并绑定弹性公网IP。
- 安全组只开必要端口。22端口用于SSH,源地址限制为办公网出口IP,不要直接设成0.0.0.0/0。Jupyter的8888端口也不建议直接暴露在公网,更稳妥的做法是通过SSH隧道访问。
四、装驱动、CUDA和PyTorch
实例启动后用SSH登录,先执行系统更新,再安装NVIDIA驱动。驱动版本要和CUDA版本配套,否则框架会识别不到显卡,排查起来很费时间。驱动装好后运行nvidia-smi,如果能看到显卡型号、显存占用和驱动版本,说明驱动已经正常工作。这一步没通过,后面的步骤先别做。
接下来安装CUDA和cuDNN,再用conda为每个项目建独立的Python环境,然后安装PyTorch或TensorFlow。一个项目一个环境,升级某个库时不会影响其他项目。装完后跑一段简单的张量运算,比如在GPU上做矩阵乘法,并确认设备是cuda而不是cpu。输出正常,才算框架真正调用到了显卡。
五、一个部署案例
某北京的中小型AI团队需要训练一个工业零件缺陷检测模型。原本他们依赖两台办公室工作站,训练一轮要接近两天。团队在华为云北京区域创建了一台单卡GPU实例,按需计费,先用一周时间完成环境搭建和小批量验证。
部署时遇到的主要问题是驱动与CUDA版本不匹配,框架调不到显卡。他们后来按官方兼容表重新安装,问题才解决。训练阶段,数据集放在云盘里,同时定时对模型权重做快照,防止中途意外中断后白跑。实际用下来,单轮训练时间明显缩短,工程师可以把精力放在调参上,不用一直等结果。
六、常见问题怎么处理
- 驱动装不上:先检查内核版本和驱动是否匹配,必要时安装与当前内核对应的kernel-devel包。内核升级之后,驱动往往也要重新处理。
- 显存溢出:先把batch size调小,不行再考虑混合精度训练。改之前记下原来的参数,方便对比效果。
- 成本过高:训练结束就关机或释放实例,数据盘单独保留。忘了关机,是最常见的隐形开销。
- 安全隐患:定期更新系统补丁,禁用root密码登录,并开启云平台的操作日志审计。出问题时,能查到是谁、什么时候做了什么。
七、搭好之后怎么走
整个流程并不复杂。前期把算力需求想清楚,创建时选对规格和镜像,再把驱动与CUDA版本对齐,大多数团队半天左右就能完成基础环境部署。等业务量上来,再考虑多卡扩展或容器化管理,那时做规划也不迟。