在Google Colab安装TensorFlow GPU遇subprocess-exited-with-error错误求助

错误原因 #

TensorFlow从2.10版本起，官方已弃用单独的tensorflow-gpu包，不再维护该包的发行版。你尝试安装的tensorflow-gpu-2.12.0仅保留了空的源码包，无法正常生成安装所需的元数据，因此触发metadata-generation-failed错误。

解决办法 #

直接安装标准tensorflow包即可，在Colab的GPU运行时环境中，该包会自动检测并启用GPU支持（Colab已预先配置好CUDA、cuDNN等GPU依赖）。

替代安装步骤 #

1. 卸载现有冲突的TensorFlow版本（如果存在）

!pip uninstall -y tensorflow tensorflow-gpu

2. 安装最新兼容的TensorFlow版本

!pip install tensorflow

3. 验证GPU是否成功启用

import tensorflow as tf
# 列出可用的GPU设备
print(tf.config.list_physical_devices('GPU'))
# 验证GPU运算可用性
print(tf.test.is_gpu_available())

若输出中包含PhysicalDevice(name='/physical_device:GPU:0', device_type='GPU')且is_gpu_available()返回True，则说明GPU配置成功。

额外注意事项 #

• Colab默认预装了适配GPU的TensorFlow版本，可先通过!pip show tensorflow查看当前版本，若版本在2.10及以上，无需重装，直接验证GPU即可。
• 若仍遇到问题，可尝试指定具体兼容版本安装，例如：!pip install tensorflow==2.15.0（该版本对T4 GPU兼容性良好）

内容的提问来源于stack exchange，提问作者Partha Pratim Sarma