如何在Google colaboratory上使用GPU升级Tensorflow [英] How to upgrade tensorflow with GPU on google colaboratory

问题描述

当前google colaboratory使用tensorflow 1.4.1.我想将其升级到1.5.0版本.每当我执行!pip install --upgrade tensorflow命令时，笔记本实例都会成功将tensorflow版本升级到1.5.0.但是升级操作后，tensorflow实例仅支持"CPU".

Currently google colaboratory uses tensorflow 1.4.1. I want to upgrade it to 1.5.0 version. Each time when i executed !pip install --upgrade tensorflow command, notebook instance succesfully upgrades the tensorflow version to 1.5.0. But after upgrade operation tensorflow instance only supports "CPU".

当我执行了此命令后，它什么也没显示:

When i have executed this command it shows nothing :

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

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

是否应该有另一种方式来升级Tensorflow?如升级到tensorflow-gpu包?笔记本什么时候会带有升级的张量流?

Should there be another way for upgrading tensorflow ? such as upgrading to tensorflow-gpu package ? Also when will notebooks will come with upgraded tensorflows ?

推荐答案

即使您要安装gpu版本!pip install tensorflow-gpu==1.5.0，由于cuda库的原因，它仍然无法导入.目前，我还没有找到在GPU上使用1.5版本的方法.因此，我宁愿使用带gpu的1.4.1而不是不带gpu的1.5.

Even if you will install gpu version !pip install tensorflow-gpu==1.5.0 it will still fail to import it because of the cuda libraries. Currently I have not found a way to use 1.5 version with GPU. So I would rather use 1.4.1 with gpu than 1.5 without gpu.

您可以向他们发送反馈(主页-发送反馈)，并希望如果有足够的人发送类似的信息，他们将更新新的gpu版本.

You can send them a feedback ( Home - Send Feedback ) and hopefully if enough people will send something similar, they will update the new gpu version.

这篇关于如何在Google colaboratory上使用GPU升级Tensorflow的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！