尝试安装tensorflow v1.4后,启用GPU的VM在“导入”时崩溃

尝试安装tensorflow v1.4后,启用GPU的VM在“导入”时崩溃,tensorflow,google-colaboratory,Tensorflow,Google Colaboratory,我试图 !!pip安装tensorflow gpu==1.4,在支持协作gpu的Python 2实例中,并收到运行时错误消息。自动重新启动错误 之后,任何导入模块的尝试都会失败: 将numpy作为np导入 导入tensorflow作为tf 导入编解码器 进口泡菜 导入时间 他死了。自动重启 我该如何解决这个问题?我无法再运行任何东西,即使我重新启动笔记本或虚拟机 我猜TensorFlow 1.4使用的Cuda库版本与实例中安装的版本不同 我认为您可以pip卸载tensorflow gpu并安装

我试图

!!pip安装tensorflow gpu==1.4,在支持协作gpu的Python 2实例中,并收到运行时错误消息。自动重新启动错误

之后,任何导入模块的尝试都会失败:

将numpy作为np导入 导入tensorflow作为tf 导入编解码器 进口泡菜 导入时间

他死了。自动重启


我该如何解决这个问题?我无法再运行任何东西,即使我重新启动笔记本或虚拟机

我猜TensorFlow 1.4使用的Cuda库版本与实例中安装的版本不同


我认为您可以pip卸载tensorflow gpu并安装更新的版本,以使导入tensorflow再次工作。

Colaboratory中当前的TF版本是1.6.0。您应该将运行时环境的类型配置为gpu

使用此功能后

执行是透明的:

如果出于某种原因您需要1.4.0版,那么Colaboratory有时会显示出一些奇怪的行为,但过程非常简单。


*如果您需要py2,请执行相同的过程,但更改语言版本。

是的,这很有效!非常感谢。我忘了CUDA图书馆在幕后也是一个依赖关系。这是一个彻底的回应-谢谢!不幸的是,我已经在一个支持Python 2 GPU的实例上运行了。问题在于我试图强制使用不同的Tensorflow版本。Super@MihirTendulkar!重要的是你解决了这个问题。祝您有个美好的一天!