web-dev-qa-db-ja.com

グーグルコラボラトリーでGPUを使用してテンソルフローをアップグレードする方法

現在、googlecolaboratoryはtensorflow1.4.1を使用しています。 1.5.0バージョンにアップグレードしたい。 _!pip install --upgrade tensorflow_コマンドを実行するたびに、ノートブックインスタンスはtensorflowバージョンを1.5.0に正常にアップグレードします。ただし、アップグレード操作後、tensorflowインスタンスは「CPU」のみをサポートします。

このコマンドを実行すると、何も表示されません。

from tensorflow.python.client import device_lib device_lib.list_local_devices()

TensorFlowをアップグレードする別の方法が必要ですか？ tensorflow-gpuパッケージへのアップグレードなど？また、ノートブックにはいつアップグレードされたテンソルフローが付属しますか？

pythontensorflowgoogle-colaboratory
3
AGP

GPUバージョンをインストールする場合でも!pip install tensorflow-gpu==1.5.0 cudaライブラリがあるため、インポートに失敗します。現在、GPUで1.5バージョンを使用する方法が見つかりません。したがって、GPUなしの1.5よりもGPUありの1.4.1を使用したいと思います。

あなたは彼らにフィードバックを送ることができます（ Home - Send Feedback ）そして、十分な数の人が同様のものを送信する場合は、新しいGPUバージョンを更新することを願っています。

2
Salvador Dali

Google Colaboratoryは、バージョン1.6.0rc1までのテンソルフローをサポートするようになりました。

import tensorflow as tf
tf.__version__

#'1.6.0-rc1'

そして、 GPUハードウェアアクセラレータを有効にする の場合、GPUは機能するはずです。次に、物事が機能しているかどうかをテストできます。

from tensorflow.python.client import device_lib
device_lib.list_local_devices()

出力：

[name: "/device:CPU:0"
 device_type: "CPU"
 memory_limit: 268435456
 locality {
 }
 incarnation: 14621691266205111434, name: "/device:GPU:0"
 device_type: "GPU"
 memory_limit: 198836224
 locality {
   bus_id: 1
 }
 incarnation: 17821632640358169265
 physical_device_desc: "device: 0, name: Tesla K80, pci bus id: 0000:00:04.0, compute capability: 3.7"]
2
weiji14

編集：これは機能しません。サルバドールの答えを見てください。

Tensorflowをアンインストールします

!pip uninstall tensorflow -y

インストールtensorflow-gpu

!pip install tensorflow-gpu==1.5.0
0
Shubham Bhardwaj