来自 TensorFlow 的 "Getting Started"页:
# Only CPU-version is available at the moment.
$ pip install https://storage.googleapis.com/tensorflow/mac/tensorflow-0.5.0-py2-none-any.whl
我对使用 GPU 或 CUDA 库不是很熟悉,但是如果我在 Linux VM 中安装了 TensorFlow(比如 precise32 可通过 Vagrant 获得),那么在该 VM 中运行时 TensorFlow 会使用 GPU 吗?
最佳答案
可能不是。 VirtualBox,例如,does not support PCI Passthrough on a MacOS host, only a Linux host (即使那样,我也会……呃,不要抱有希望)。 MacOS 最终与其 GPU 紧密集成,以至于我非常怀疑此时任何 VM 都可以做到这一点。
关于macos - 如果 TensorFlow 安装在 VM 上,它会在 Mac 上使用 GPU 吗?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/33656689/