目前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(使用前将#替换为@)