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

Question

我尝试过

!pip install tensorflow-gpu==1.4 在 Colaboratory（启用 GPU 的 Python 2 实例）中并收到“运行时死亡。自动重启”错误。

之后，任何导入模块的尝试都会失败：

import numpy as np import tensorflow as tf import codecs import pickle import time

...

Runtime died. Automatically restarting

我该如何解决这个问题？即使我重新启动笔记本或重新启动 VM，我也无法再运行任何东西。

Answer 1

我的猜测是 TensorFlow 1.4 构建的 Cuda 库版本与实例中安装的库版本不同。

我认为您可以pip uninstall tensorflow-gpu 并安装更新版本以使import tensorflow 再次工作。

Answer 2

目前 Colaboratory TF 版本为 1.6.0 需要将运行环境类型配置为 GPU。

使用此功能后，

执行是透明的：

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

*如果你需要py2，同样的过程，但改变语言的版本。