GPU服务器对比普通服务器,核心差别就是搭载多块独立显卡。这类硬件可以承载深度学习训练、科学运算、视频转码,也能用于挖矿,但新手拿到机器,往往不知道怎么登录操作。下面梳理从采购、上线到硬件选型的关键要点,尽量精简表述。
拿到GPU服务器,第一步就是远程连接。和普通云服务器一致,SSH是主流连接方式。Windows可以用Xshell或者PowerShell,Mac、Linux直接打开终端。连接命令为ssh 用户名@IP地址,输入密码登录。服务商提供密钥文件的情况下,使用ssh -i 密钥文件路径 用户名@IP。成功连接不要立刻跑任务,先执行nvidia-smi。这条命令会展示所有GPU型号、显存占用、温度、功耗和驱动版本。能够正常列出显卡,代表驱动安装完成,硬件环境可用。如果提示command not found,说明驱动未安装或者CUDA环境缺失,需要优先处理。

nvidia-smi输出中的CUDA Version,代表驱动支持的最高CUDA版本,不等于当前系统实际安装版本。查看本地CUDA版本,执行nvcc -V。两组数字不一致很常见,驱动版本高于CUDA版本属于正常现象。运行PyTorch时,torch.cuda.is_available()返回True,GPU才算就绪。返回False,大多是PyTorch的CUDA版本和系统不匹配,重新安装torch即可,直接去PyTorch官网复制适配CUDA版本的安装命令,不要自行猜测版本号。
数据上传到服务器,小文件用scp,大容量文件优先rsync,数据集体量巨大可以挂载对象存储或者从网盘拉取。代码上传完成,长时间运行任务记得用tmux或者screen新建会话,SSH断开连接任务也不会终止。tmux基础操作简单,tmux new -s train创建名为train的会话,在会话内执行任务,按下Ctrl+B再按D脱离会话,后续用tmux attach -t train恢复查看。省略这一步,网络波动就会中断任务,耗费大量时间白跑。
租用GPU服务器最容易忽略成本管控。按量计费实例,闲置时依旧持续扣费,任务结束及时关机,或者配置闲置自动释放策略。任务允许中断,像批量推理、参数调试这类场景,选择竞价实例,价格一般为按量付费的三到五折,缺点是资源紧张时实例会被回收。任务运行前配置好checkpoint,被回收后可以从断点继续训练。存储同样持续计费,实例关机后数据盘依旧扣费,无用数据及时清理。
硬件选型阶段,GPU型号是首要选择。RTX 4090是个人与小型团队常用型号,24GB显存,单精度、半精度算力满足多数需求,价格远低于专业显卡。A100、H100属于数据中心级显卡,显存容量大,NVLink互联带宽高,适合多卡训练大模型,单卡采购成本可以购置多台4090整机。选型参考模型规模,参数量10B以内,4090基本够用;达到50B甚至100B级别,才需要考虑A100这类高端卡。
显存容量比核心数量更加关键。任务跑不起来,很多时候不是算力不足,而是显存空间不够。推理场景,模型权重、中间激活值、KV缓存都会占用显存。粗略估算,FP16精度下,每10亿参数大概占用2GB显存。7B模型仅权重就需要14GB,叠加推理缓存,16GB显存勉强运行,24GB会更宽裕。训练场景显存需求会成倍增加,梯度、优化器参数、激活值都要占用显存。挑选显卡优先保证显存充足,避免运行中途触发OOM报错。
CPU和内存搭配经常被忽视。GPU服务器不能只看显卡,CPU性能偏弱,数据预处理和加载会形成瓶颈,显卡空闲等待数据输入。常规配比,单块GPU搭配8至16个CPU核心相对均衡,内存容量至少为显存的1.5到2倍。举例,4块24GB的4090,总显存96GB,系统内存最好配置192GB以上。PCIe通道数量也要留意,多卡训练时GPU之间通信依靠PCIe或者NVLink,通道不足会拖慢梯度同步速度。
存储层面,NVMe SSD属于标配。机械硬盘不适合深度学习场景,数据读取速度过低。系统盘选用小容量NVMe,数据盘使用大容量NVMe或者RAID阵列。数据集体量达到数TB图像、视频文件,可以考虑网络存储或者分布式文件系统,本地磁盘只存放高频使用的热数据。
网络带宽影响多机多卡训练效率。单机多卡依靠NVLink或者PCIe通信,跨机器通信使用以太网或者InfiniBand。万兆以太网是最低标准,条件允许升级25G或者100G。InfiniBand多用于高性能计算场景,延迟低、带宽高,但网卡与交换机采购成本高,普通深度学习训练,万兆以太网就可以满足需求。
自建GPU服务器,电源与散热很容易出问题。单张RTX 4090功耗450W,四张合计1800W,叠加CPU、主板功耗,整机功耗轻松突破2500W。电源预留充足余量,选择80 Plus金牌及以上规格,额定功率2000W起步。散热方面,涡轮卡更适合多卡密集部署,涡轮卡直接把热风排出机箱,不会在机箱内部循环积热。机箱风道做好规划,前进后出或者下进上出,避免热风在显卡之间循环堆积。
最后简单核算成本。一台搭载4张4090的工作站,硬件成本大约五六万。租用云服务器同等配置按量计费每小时十几元,不间断运行一个月花费几千。仅偶尔做实验,租赁方案更划算。长期运行、数据保密要求高,或是需要定制硬件,自建工作站更合适。不用直接选购顶配设备,先把业务流程跑通,找到性能瓶颈再逐步升级硬件。