如果安装在VM上,TensorFlow会在Mac上使用GPU吗?

时间:2015-11-11 17:50:32

标签: macos cuda virtual-machine gpu tensorflow

来自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(比如通过Vagrant提供的precise32),那么TensorFlow会在内部运行时使用GPU VM?

2 个答案:

答案 0 :(得分:6)

可能不是。 VirtualBox,例如,does not support PCI Passthrough on a MacOS host, only a Linux host(即便如此,我......呃,不要让我的希望得到)。 MacOS最终与其GPU紧密集成,我非常怀疑任何虚拟机都能在这一点上做到这一点。

答案 1 :(得分:4)

作为更新:Tensorflow现在可以在Mac OS X上使用GPU。相关的PR为https://github.com/tensorflow/tensorflow/pull/664,并且在brew install coreutils Linux安装之后'来自源代码'说明应该有效。与使用NVIDIA gforce 960和Intel i7-6700K的CPU版本相比,我看到了10倍的加速。