Numba 的安装是否会干扰 tensorflow 并扰乱 cuda 安装？

Question

我想知道在 python2.7 中安装 Numba 是否会干扰在 Ubuntu 16.04 上安装（并正常工作）的 tensorflow-gpu？我想使用 numba 对向量和矩阵（例如向量化）进行 gpu 加速计算，然后在 tensorflow 深度学习模型中使用这些计算的矩阵。

Answer 1

正如@Robert Crovella 在评论中正确提到的那样，当您创建会话时，Tensorflow 将尝试分配它可以分配的最大内存量。如有必要，您可以避免这种情况的发生：

一般来说，在同一个 GPU 上 运行 多个进程是完全没问题的。主要问题只是内存分配。如果 Numba 预先分配它需要的内存，你就不会有问题（免责声明我不太了解 Numba）。如果 Numba 需要在您的 tensorflow 会话分配后分配内存，则需要遵循上面的 link 以避免内存分配问题。

我相当确定您不会在系统之间遇到任何类型的 CUDA 驱动程序冲突。至少我从来没有听说过这是一个问题，而且我从 v0.12 开始每天都在使用 tensorflow。