tf.test.is_gpu_available（）在GCP上返回False

Question

我正在使用Tesla V100在GCP的筆記本電腦上訓練CNN。 我已經在自己的自定義數據上訓練了一個簡單的yolo，雖然速度很快，但不太准確。 因此，我決定從頭開始編寫自己的代碼，以解決我要解決的問題的特定方面。

在GCP之前，我曾嘗試在Google Colab上運行我的代碼，效果很好。 Tensorflow可以檢測到GPU，並且無論是Tesla K80還是T4都可以使用它。

from tensorflow.python.client import device_lib
print(device_lib.list_local_devices())
tf.test.is_gpu_available()  #>>> True

我的問題是，這個相同的函數在GCP筆記本上返回False ，就好像Tensorflow無法使用在GCP VM上檢測到的GPU一樣。 我不知道有任何命令會迫使Tensorflow在CPU上使用GPU，因為它會自動執行。 我已經嘗試安裝或卸載然后再安裝某些版本的tensorflow，tensorflow-gpu和tf-nightly-gpu（例如1.13和2.0dev），但是它沒有任何效果。

nvidia-smi的輸出

Answer 1

您是否嘗試過使用GCP的AI平台筆記本 ？ 他們提供預配置有Tensorflow的VM，並已安裝所有必需的GPU驅動程序。