MAC安装 stable diffusion Torch not compiled with CUDA enabled
时间: 2025-01-03 11:35:50 浏览: 21
### 解决方案
对于希望在 Mac 上安装支持 CUDA 的 PyTorch 来运行 Stable Diffusion 遇到的问题,需要注意的是当前大多数 Mac 设备并不直接支持 NVIDIA GPU 和 CUDA 技术。然而,如果用户拥有配备有 NVIDIA 显卡的较旧款外接显卡 (eGPU) 或者通过虚拟机方式连接外部带有NVIDIA GPU的工作站,则可以尝试配置支持 CUDA 的环境。
由于 Apple Silicon 架构(M1/M2 芯片系列)不兼容传统的基于 NVIDIA 的 CUDA 库,因此官方并没有提供针对这些设备的支持CUDA版本PyTorch[^3]。但是为了实现类似功能,开发者们转向了 Metal Performance Shaders (MPS),这是苹果公司专门为 macOS 提供的一种高性能图形处理API,在某些情况下能够加速机器学习模型训练过程中的计算任务。
#### 使用 MPS 支持来替代 CUDA 进行开发:
考虑到上述情况,建议采用如下方法设置适合于 Mac 平台上的 Stable Diffusion 开发环境:
1. **确认 Python 版本**: 确认已安装最新稳定版 Python 及 pip 工具;
2. **创建并激活虚拟环境**:
```bash
python3 -m venv myenv
source myenv/bin/activate # Linux/macOS
```
3. **安装依赖项**
更新 `pip` 到最新版本,并安装必要的包:
```bash
pip install --upgrade pip setuptools wheel
```
4. **获取适用于 MacOS 的 PyTorch 发布版本**, 包含对 MPS 加速器的支持:
```bash
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
```
对于想要利用Metal框架性能优化的情况,请确保选择了包含 `-mps` 后缀标记的二进制文件下载链接。
5. **克隆所需仓库并按照说明完成其余组件部署**
6. **验证安装成功与否**:
测试 Torch 是否能识别 MPS 设备作为默认张量运算平台之一。
```python
阅读全文