如何在Colab本地运行时调用Apple M1本地GPU？

问题根源 #

nvcc和nvidia-smi是NVIDIA GPU专属工具，M1 Pro采用Apple Silicon架构（基于Metal加速），因此这些命令完全不适用，需替换为适配Apple芯片的检查与调用方式。

步骤1：替换GPU硬件检查命令 #

用以下命令查看M1 Pro的硬件信息，替代原NVIDIA专属命令：

# 查看CPU型号（确认是Apple Silicon）
!sysctl machdep.cpu.brand_string
# 查看GPU芯片信息
!system_profiler SPDisplaysDataType | grep "Chipset Model"

步骤2：验证框架与Cellpose的GPU识别 #

M1 Pro的GPU加速依赖Metal Performance Shaders（MPS）后端，需确保PyTorch/TensorFlow及Cellpose支持该后端。修改原代码为：

import os
import torch
from cellpose import core

# 先验证PyTorch对MPS后端的支持
print(f"PyTorch MPS后端可用: {torch.backends.mps.is_available()}")
print(f"PyTorch编译时已包含MPS: {torch.backends.mps.is_built()}")

# 检查Cellpose是否启用GPU加速
# Cellpose会自动检测MPS后端，若未触发可手动设置环境变量
os.environ['CELLPOSE_MPS'] = '1'
use_GPU = core.use_gpu()
yn = ['NO', 'YES']
print(f'>>> GPU activated? {yn[use_GPU]}')

关键注意事项 #

• 本地Python环境必须是ARM64架构版本（避免用Rosetta转译的x86版本），推荐从Python官网下载或用conda-forge安装。
• 确保Cellpose更新至最新版本，旧版本可能未适配Apple Silicon：

  !pip install --upgrade cellpose
  

• 若仍无法启用GPU，需确认PyTorch为M1适配版本：

  !pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
  

  （注：PyTorch的M1版本会自动识别并启用MPS，无需单独指定GPU包）

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