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