在无NVIDIA GPU的Google Colab中如何设置Facebook SAM的device？

核心修改方案 #

直接将代码中的device = "cuda"替换为device = "cpu"，修改后的完整代码如下：

import sys
sys.path.append("..")
from segment_anything import sam_model_registry, SamAutomaticMaskGenerator, SamPredictor

sam_checkpoint = "sam_vit_h_4b8939.pth"
model_type = "vit_h"

device = "cpu"

额外说明 #

• 切换到CPU运行后，模型推理速度会显著变慢，尤其是vit_h这类大模型。如果是在Colab环境，可尝试启用免费GPU资源：点击顶部菜单栏「Runtime」→「Change runtime type」，在弹出窗口中选择「Hardware accelerator」为「GPU」并保存，之后就能继续使用device = "cuda"。
• 若只能用CPU，可考虑更换为更小的模型（比如vit_b或vit_l），能一定程度提升推理效率。

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