我在安装了 CPython 的 Visual Studio 中设置了 Python 工具。
在 Visual Studios 中,如果我运行以下代码:
print("hello");
import numpy;
print("hello");
程序运行正常,打印两个'hello',正常退出。
但是,如果我运行以下代码:
print("hello");
import tensorflow;
print("hello");
程序挂起,打印一个“你好”,并拒绝继续。
所有包都应该正确安装 - 在 Python 交互窗口中使用 TensorFlow 打印正确的输出并完美运行。
为什么第二种情况程序会挂掉?
最佳答案
一旦你导入 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 版本。
关于python - 导入 TensorFlow 会停止运行 python 程序,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/41427951/