簡體   English   中英

如何在 Google colab 中更改 Keras/tensorflow 版本?

[英]How to change Keras/tensorflow version in Google colab?

我在 google colaboratory 上使用 keras/tensorflow,我需要回到它們的以前版本。

問題是當我運行!pip install q keras==1.2.2 ,內核顯示安裝了 keras 1.2.2 但是當我使用keras.__version_檢查它時它顯示 2.1.6 。 tensorflow 也是同樣的情況。

我該如何解決?

我昨天有這個問題。 我很驚訝安裝軟件包沒有效果。 但是我意識到我需要重新啟動內核。 在 Colab 中,這稱為Restart runtime 重新啟動后,新版本應該可供您使用。

在這里你可以找到重啟:

在此處輸入圖片說明

您的命令!pip install q keras==1.2.2 for Keras 沒問題,只是您需要像其他答案提到的那樣重新啟動運行時。
但是對於 tensorflow 你應該使用%tensorflow_version 1.x
正如他們在這里所說:

避免對 GPU 和 TPU 使用 pip install 我們建議不要使用 pip install 為 GPU 和 TPU 后端指定特定的 TensorFlow 版本。 Colab 從源代碼構建 TensorFlow,以確保與我們的加速器組兼容。 通過 pip 從 PyPI 獲取的 TensorFlow 版本可能會遇到性能問題或根本無法工作。

正如它所說,當我嘗試用 pip 將我的 TensorFlow 降級到 1.5 時,它無法識別 GPU,只是用 cpu 訓練了我的模型。

在最新的 colab 中,當您升級或降級模塊時,您會在輸出單元格中自動看到以下語句:

“警告:以前在此運行時中導入了以下包:[scipy] 您必須重新啟動運行時才能使用新安裝的版本。

" 要求您重新啟動運行時及其按鈕。

暫無
暫無

聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.

 
粵ICP備18138465號  © 2020-2024 STACKOOM.COM