python - 导入 TensorFlow 会停止运行 python 程序

Question

我在安装了 CPython 的 Visual Studio 中设置了 Python 工具。

在 Visual Studios 中，如果我运行以下代码:

print("hello");
import numpy;
print("hello");

程序运行正常，打印两个'hello'，正常退出。

但是，如果我运行以下代码:

print("hello");
import tensorflow;
print("hello");

程序挂起，打印一个“你好”，并拒绝继续。

所有包都应该正确安装 - 在 Python 交互窗口中使用 TensorFlow 打印正确的输出并完美运行。

为什么第二种情况程序会挂掉？

Answer 1

一旦你导入 tensorflow，它会自动尝试加载 cuda，它会打印如下内容:

I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcublas.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcudnn.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcufft.so locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcuda.so.1 locally
I tensorflow/stream_executor/dso_loader.cc:128] successfully opened CUDA library libcurand.so locally

所以我认为正在发生的事情是您没有正确安装 cuda，因此它失败了。您可以尝试安装不使用 GPU 且不加载这些库的 CPU 版本。