859 字
4 分钟
在macOS中配置深度学习环境

流程#

flowchart TD A[开始] --> B{使用 Zsh} B -- 是 --> C[安装 Homebrew] B -- 否 --> D(设置 Zsh) C --> E[安装 Anaconda] D --> C E --> F[安装 VSCode] F --> G[配置 Python 虚拟环境] G --> H[安装 PyTorch] H --> I[完成] style A fill:#f96,stroke:#333,stroke-width:2px style I fill:#f96,stroke:#333,stroke-width:2px

使用 Zsh#

macOS默认的终端就是Zsh

安装 Homebrew#

打开终端,使用命令行检查是否已经安装了 Homebrew:

brew --version

如果看到 zsh: command not found: brew 的错误信息。只需在终端中运行以下命令即可开始安装:

/bin/bash -c "$(curl -fsSL https://raw.githubusercontent.com/Homebrew/install/HEAD/install.sh)"

安装完成后,建议运行 brew doctor 检查一切是否正常。

安装 Anaconda#

对于 Anaconda,可以通过以下命令安装:

brew search --cask anaconda brew install --cask anaconda 看到下面的话意味着安装成功: PREFIX=/opt/homebrew/anaconda3 Unpacking payload ... Installing base environment... Preparing transaction: ...working... done Executing transaction: ...working...  done installation finished. ==> **Changing ownership of paths required by anaconda with sudo; the password may** 🍺  anaconda was successfully installed!

输入conda -V或者conda --version 可能看到 zsh: command not found: conda 的错误信息。这是因为没有声明全局变量导致终端不识别安装的软件等命令,需要进行配置。根据PREFIX=/opt/homebrew/anaconda3配置:

echo 'export PATH="/opt/homebrew/anaconda3/bin:$PATH"' >> ~/.zshrc source ~/.zshrc cat ~/.zshrc

此时输入conda -V查看得conda 24.9.2,只要有版本号就是对的。

然后换源,因为不在国外,使用官方默认的源可能会有不好的网络下载体验,所以要换成国内的镜像加速源,比如清华源。

默认源:

conda config --show channels channels:   - https://repo.anaconda.com/pkgs/main   - https://repo.anaconda.com/pkgs/r

添加清华源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main conda config --set show_channel_urls yes

会得到:

conda config --show channels channels:   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main   - https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free   - https://repo.anaconda.com/pkgs/main   - https://repo.anaconda.com/pkgs/r

安装VSCode#

官网下载适用于macOS版本的。

安装完后记得下载适用于python的插件:

Chinese (Simplified) # 简体中文 Python # 包含Python、Pylance、Python Debugger的整合包 autopep8 # 用来格式化python代码

配置 Python 虚拟环境#

比如在我在vscode中创建了一个名字为torch_mps_py310的虚拟环境:

conda create --name torch_mps_py310 python=3.10

然后我在右下角的选择python解释器中选择它,重新打开一个终端,它就被激活了。

默认包:

pip list Package Version ---------- ------- pip 25.0 setuptools 75.8.0 wheel 0.45.1

默认源:

pip config list # 返回如下: global.index-url='https://mirrors.aliyun.com/pypi/simple/'

我的默认是阿里源,就不换了。如果返回是空的,请换源🙏 。 指令如下:

pip config set global.index-url='https://mirrors.aliyun.com/pypi/simple/'

更新一下pippip install --upgrade pip 或者 python -m pip install --upgrade pip(权限不够使用这个)

安装 PyTorch#

去到官网 选择Stable->Mac->Pip->Python->Default,复制命令并在vscode终端激活的torch_mps_py310环境中运行:

pip3 install torch torchvision torchaudio

安装完后大概有这些包:

Package Version ----------------- -------- filelock 3.17.0 fsspec 2025.2.0 Jinja2 3.1.5 MarkupSafe 3.0.2 mpmath 1.3.0 networkx 3.4.2 numpy 2.2.2 pillow 11.1.0 pip 25.0 setuptools 75.8.0 sympy 1.13.1 torch 2.6.0 torchaudio 2.6.0 torchvision 0.21.0 typing_extensions 4.12.2 wheel 0.45.1

测试是否成功安装#

import torch # 查看 torch安装是否成功 并查看其版本 print(torch.__version__) # 查看 mps是否安装成功 是否可用 print(torch.backends.mps.is_available()) # 检查 GPU 是否可用 print(torch.cuda.is_available()) # 对于 CUDA, 返回 False 是正常的 print(torch.backends.mps.is_available()) # 对于 MPS, 应该返回 True # 获取 MPS 设备 mps_device = torch.device("mps") print(mps_device) # 输出 "mps"

返回下面是正确的

2.6.0 True False True mps

附录#

硬件配置#

硬件组件配置详情
设备型号Mac mini
芯片Apple M2
内存8GB
存储空间256GB
操作系统版本macOS Ventura 13.4.1(c)

对于conda的选择#

请参考这篇文章

在macOS中配置深度学习环境
https://calabchen.github.io/posts/ai/index1/
作者
半入烟云
发布于
2025-02-01
许可协议
CC BY-NC-SA 4.0