Skip to main content
 首页 » 编程设计

python之无法在 Google Colab 上将 GPU 用于 tensorflow 2.0

2024年09月03日26emanlee

我尝试在 Google Colab 上使用 GPU 运行我的笔记本,但它没有为我提供 GPU,但是当我使用 tensorflow 1.15.0 运行笔记本时,GPU 可用。

tf.test.gpu_device_name() 

给出输出 '/device:GPU:0' 对于 tensorflow 1.15.0

但是当我对 tensorflow 2.0.0 执行相同操作时,该函数返回 ''。

请您参考如下方法:

我解决了在 google colab 中安装

!pip install tensorflow-gpu  

和

!pip install tf-nightly  

所以现在tf.test.gpu_device_name(),输出是/device:GPU:0 但是,TensorFlow 自动将其版本升级到 2.1.0-dev20191120