Mac 安装 PyTorch

一.基本概念PyTorch 是 Meta原 Facebook开源的 深度学习框架用 Python 写模型、训练模型、做推理。在研究和工业里都很常见特点是 Python 风格自然、动态计算图、调试方便。PyTorch 是一个专为深度学习优化的张量计算库支持在 GPU 和 CPU 上进行高效运算。官网地址https://pytorch.org/PyTorch 文档: https://docs.pytorch.org/Mac 上的 PyTorch 特点Mac 上一般选 CPU 或 MPS不要选 CUDA 版本。注意最新稳定版 PyTorch 需要 Python 3.10 或更高版本。二.安装1.确认芯片类型uname-marm64 → Apple SiliconM 系列可用 MPS 加速x86_64 → Intel Mac只能用 CPU2.确认Python版本python3--version要求Python3.103.创建虚拟环境# 进入项目目录cd/path/to/your/project# 创建虚拟环境pytorch-envpython3-mvenv pytorch-env# 激活虚拟环境进入虚拟环境sourcepytorch-env/bin/activate# 升级 pippipinstall--upgradepip4.安装Pytorch直接执行pip3installtorch torchvision torchaudio三.验证是否安装成功看版本python-cimport torch; print(torch.__version__)看 MPS 是否可用Apple Siliconpython-cimport torch; print(MPS available:, torch.backends.mps.is_available())输出 MPS available: True 表示可用 GPU 加速。简单跑一下张量importtorch# CPUxtorch.randn(3,3)print(x)# Apple Silicon 可试 MPSiftorch.backends.mps.is_available():devicetorch.device(mps)ytorch.randn(3,3,devicedevice)print(MPS tensor:,y)保存为 test_torch.py运行python test_torch.py四. 和 Jupyter 一起用# 装 Jupyter 和内核支持pipinstalljupyter ipykernel# 注册内核名字可自定python-mipykernelinstall--user--namepytorch-env --display-namePython (PyTorch)# 启动 Jupyterjupyter notebook当前状态已成功启动关键几行Serving notebooks fromlocaldirectory: /Users/TechFish/pythonproject/pytorchtest Jupyter Server2.20.0 is running at: http://localhost:8888/tree?token...含义Jupyter 已在后台运行工作目录是/Users/TechFish/pythonproject/pytorchtest本机访问地址http://localhost:8888token… 是本次启动的访问令牌防止别人随便连你的本机服务然后在 Jupyter 里选 pytorch-env 内核。在 Notebook 里Kernel → Change Kernel选 Python (PyTorch)。如何确认注册成功查看已注册内核jupyter kernelspec list示例输出Available kernels: python3 /usr/local/share/jupyter/kernels/python3 pytorch-env /Users/你的用户名/Library/Jupyter/kernels/pytorch-env在 Notebook 里验证新建单元格运行importsys print(sys.executable)importtorch print(torch.__version__)sys.executable 应指向你的 虚拟环境名/bin/python打开Jupyter 界面后怎么用:1.在文件列表页点 New → Notebook2.若装了多个内核选 pytorch-env或你注册的显示名Python (PyTorch)–namepytorch-env --display-name“Python (PyTorch)”3.在单元格里测试importsysprint(sys.executable)importtorchprint(torch.__version__)print(MPS:,torch.backends.mps.is_available())4.按 Shift Enter 运行当前单元格5.停止服务在该终端按 Ctrl C可能要按两次确认五.安装后建议把依赖记下来方便复现pip freezerequirements.txtrequirements.txt 里会有类似torch2.x.xtorchvision0.x.xtorchaudio2.x.x