我可以在 Radeon Pro 450/455 上使用 GPU 加速运行深度学习框架(Theano 或 Tensorflow)吗?

我可以在 Radeon Pro 450/455 上使用 GPU 加速运行深度学习框架(Theano 或 Tensorflow)吗?

我正在考虑买一台新笔记本电脑。对于我的工作来说,如果能在上面运行 GPU 加速的神经网络框架,尤其是 Theano 或 Tensorflow,那就太理想了。

Apple 推出了一款 Macbook Pro,其显卡可选配“Radeon Pro 450”或“Radeon Pro 460”。我能在这个 GPU 上运行 Theano 或 Tensorflow 吗?如果不能,那么其​​他框架怎么样?

我相信 Theano 和 Tensorflow 都使用/需要 CUDA 才能工作。

答案1

我有一台配备 Radeon Pro 460 的 Macbook Pro,并尝试使用具有 OpenCL 分支的 Caffe 训练分类器。根据我的经验,在 GPU 上训练模型的速度与在 CPU 上训练的速度相同!我也尝试在配备 Nvidia 960 的 Mac 上做同样的事情,你猜怎么着;它快了 40 倍!CUDA 使用经过优化的结构进行繁重的矩阵计算,不幸的是,AMD 显卡无法提供这种结构。因此,如果您的目标是运行基于 GPU 的 NN,我个人的建议是选择配备 Nvidia Graphics 的显卡。

相关内容