步骤 1:安装 Miniforge(推荐)
1、下载 Miniforge:
https://github.com/conda-forge/miniforge
2、安装 Miniforge:
chmod +x ~/Downloads/Miniforge3-MacOSX-arm64.sh
sh ~/Downloads/Miniforge3-MacOSX-arm64.sh
3、重启终端,确认安装成功:
conda --version
步骤 2:创建并激活虚拟环境
1、创建新环境(建议 Python 3.8/3.9):
conda create -n tf_arm python=3.9
conda activate tf_arm
步骤 3:安装 TensorFlow 和 GPU 加速
1、安装 tensorflow-macos
(苹果官方优化版本):
conda install -c apple tensorflow-deps
python3 -m pip install tensorflow-macos
2、安装 GPU 加速插件 tensorflow-metal
:
python3 -m pip install tensorflow-metal
3、使用Jupyter Lab,安装常用的packages
pip install jupyterlab
步骤 4:验证安装
1、打开终端
source /Users/xujun/miniforge3/bin/activate
conda activate tf_arm
jupyter lab
运行以下代码检查 TensorFlow 和 GPU 支持:
import tensorflow as tf
print(tf.__version__) # 应输出 2.x.x
print("GPU 可用:", tf.config.list_physical_devices('GPU')) # 应显示 GPU 设备
原创文章(本站视频密码:66668888),作者:xujunzju,如若转载,请注明出处:https://zyicu.cn/?p=19882