尝试安装tensorflow v1.4后，启用GPU的VM在`import`崩溃

Question

我试图

Colaboratory中的

CURSOR c1 IS SELECT DISTINCT(DB.TABLE_NAME) FROM DBA_TABLES DB, USER_TABLES UT WHERE DB.TABLE_NAME = UT.TABLE_NAME AND DB.IOT_TYPE IS NOT NULL; BEGIN OPEN c1; FETCH c1 INTO IOT_TABLE; CLOSE c1; （支持GPU的Python 2实例）并收到“运行时死机。自动重启”错误。

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

!pip install tensorflow-gpu==1.4

...

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

我该如何解决这个问题？即使我重新启动笔记本电脑或重新启动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，相同的程序，但改变语言的版本。