【问题标题】:Can pylearn2 and Theano run on AMD GPU based platform?pylearn2 和 Theano 可以在基于 AMD GPU 的平台上运行吗?
【发布时间】:2015-12-15 21:04:15
【问题描述】:

我想使用 pylearn2、theano 和 scikit-neuralnetwork 来构建神经网络模型。但是我的朋友告诉我,所有这些模块只能在基于 NVIDIA GPU 的平台上运行(因为他们会导入 pycuda 模块)。但我只有一个 AMD GPU(R9 270,和一个 AMD FX-8300 CPU),我希望利用 AMD GPU 来加速计算。我可以使用上面提到的所有模块吗?或者有什么替代品可以用来构建神经网络模型?谢谢!

【问题讨论】:

    标签: cuda scikit-learn opencl neural-network theano


    【解决方案1】:

    目前,Theano 仅支持 nvidia GPU。有支持 AMD GPU 的 OpenCL 后端的部分实现,但它不完整且不受支持。

    scikit-neuralnetwork 基于 PyLearn2 构建,PyLearn2 构建在 Theano 之上,因此这些软件包都不能在 AMD GPU 上运行。

    Torch 似乎已经有一些OpenCL support。 Caffe 的 OpenCL 支持似乎是 under development

    【讨论】:

    • 谢谢,丹尼尔。有点可惜。 DeepCL 怎么样?与上述软件包相比,它的性能如何?还有convnetjs,它似乎不依赖GPU。 C++或者js绑定也可以。
    • Theano 在 CPU 上运行良好,如果你觉得可以的话。我对其他软件包没有足够的经验来评论它们。
    猜你喜欢
    • 1970-01-01
    • 1970-01-01
    • 2012-10-01
    • 1970-01-01
    • 2015-03-14
    • 2016-02-04
    • 2021-05-04
    • 1970-01-01
    • 1970-01-01
    相关资源
    最近更新 更多