SageMaker Studio 实验室，“制作 AI 生成图像”实验室，错误“在您的系统上找不到 NVIDIA 驱动程序”

Question

在本实验室管道建设期间

(进口手电筒 从扩散器导入 StableDiffusionPipeline

管道 = StableDiffusionPipeline.from_pretrained( “runwayml/stable-diffusion-v1-5”，torch_dtype=torch.float16，修订版=“fp16” )

管道 = pipeline.to(“cuda”)

我收到以下运行时错误：

运行时错误：在您的系统上找不到 NVIDIA 驱动程序。请检查您是否拥有 NVIDIA GPU 并从 http://www.nvidia.com/Download/index.aspx

安装了驱动程序

有解决这个问题的提示吗？

提前致谢

我是一个新来者，尝试了在以下位置找到的命令，但没有效果 https://pytorch.org/get-started/previous-versions/

Answer 1

您将希望为 Studio Lab 使用基于 GPU 的实例，您在 Studio Lab 环境背后使用的实例是什么？