SageMaker Studio Lab生成AI图像报错：系统未找到NVIDIA驱动求解决

核心原因 #

SageMaker Studio Lab默认可能未分配GPU实例，或是当前运行环境为CPU实例，调用to("cuda")时自然会触发驱动找不到的错误。

具体解决思路 #

• 切换GPU运行时（优先方案）

  1. 点击顶部菜单栏的「Runtime」→「Change runtime type」
  2. 在弹出窗口中，将「Hardware accelerator」选为「GPU」，点击「Save」
  3. 重启运行时（Runtime→Restart runtime），之后重新执行代码

• 临时适配CPU运行（速度较慢）
  若暂时无法切换GPU，可修改代码适配CPU环境：
  将torch_dtype=torch.float16改为torch_dtype=torch.float32，同时把pipeline = pipeline.to("cuda")替换为pipeline = pipeline.to("cpu")，即可在CPU上运行，但生成图片的速度会大幅降低。

• 验证GPU可用性
  先运行以下代码确认当前环境是否已分配GPU：

  import torch
  print(torch.cuda.is_available())
  

  输出True则说明GPU环境正常，可重新执行Pipeline代码；输出False则需再次检查运行时设置。

内容的提问来源于stack exchange，提问作者franck1961