如何在google colaboratory上使用GPU升级tensorflow

2024-05-27

目前google colaboratory使用tensorflow 1.4.1。我想升级到1.5.0版本。每次当我执行时!pip install --upgrade tensorflow命令,notebook实例成功将tensorflow版本升级到1.5.0。但升级后运行tensorflow实例仅支持“CPU”。

当我执行此命令时,它什么也不显示:

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

是否应该有另一种方法来升级张量流?例如升级到tensorflow-gpu包?另外,笔记本电脑什么时候会配备升级的张量流?


截至 2020 年,Colab 可以运行 Tensorflow 2.0。卸载当前版本的 Tensorflow:

!pip uninstall tensorflow

并简单地使用 pip 重新安装。

!pip install tensorflow==2.0.0

如果您选择,您还可以安装测试版

!pip install tensorflow==2.0.0-beta1

重新安装 Tensorflow 后,请确保不要忘记在“运行时”选项卡下重新启动运行时。

本机解决方案是使用以下代码启动您的代码

%tensorflow_version 2.x
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)

如何在google colaboratory上使用GPU升级tensorflow 的相关文章

随机推荐