Mac下Miniconda更新TensorFlow环境遇cudatoolkit/cudnn找不到问题求助

错误原因 #

cudatoolkit=11.2.2和cudnn=8.1.0是NVIDIA CUDA生态的专属依赖库，Mac系统（无论Intel还是Apple Silicon芯片）均不支持NVIDIA CUDA，因此conda无法在Mac的软件源中找到这两个包，导致环境解析失败。

修复步骤 #

1. 修改tools.yml配置文件
   直接移除文件中cudatoolkit=11.2.2和cudnn=8.1.0这两行依赖，修改后的文件内容如下：

channels:
    - conda-forge
dependencies:
    - jupyter
    - scikit-learn
    - scipy
    - pandas
    - pandas-datareader
    - matplotlib
    - pillow
    - tqdm
    - requests
    - h5py
    - pyyaml
    - flask
    - boto3
    - pip
    - pip:
        - tensorflow
        - bayesian-optimization
        - gym
        - kaggle

2. 重新执行环境更新命令
   在已激活的TensorFlow虚拟环境中，再次运行：

conda env update --file tools.yml

3. 验证安装有效性
   激活环境后，运行以下Python代码确认TensorFlow正常工作：

import tensorflow as tf
# 打印TensorFlow版本
print(tf.__version__)
# 检查设备情况（Apple Silicon会显示Metal设备，Intel则显示CPU）
print(tf.config.list_physical_devices())

额外注意事项 #

• 若使用Apple Silicon芯片的Mac，确保安装的是arm64架构的Miniconda版本，避免x86架构带来的兼容性问题。
• TensorFlow在Mac上的GPU加速依赖Apple Metal框架，无需手动安装任何CUDA相关组件，TensorFlow会自动启用Metal后端（Apple Silicon）或使用CPU（Intel）。

内容的提问来源于stack exchange，提问作者ethan4024