我如何使用AMD RX Radeon与mxnet软件包在R特别



我读到了将NVidia与mxnet软件包一起使用的可能性,但我没有发现任何关于使用RX Radeon GPU的信息。我可以在R中使用带有mxnet包的RX Radeon吗?以下是解释使用NVidia安装mxnet的链接:NVidia 安装

Installing MXNet on a Computer with a GPU Processor
To install MXNet R package on a computer with a GPU processor, you need the following:
Microsoft Visual Studio 2013
The NVidia CUDA Toolkit
The MXNet package
CuDNN (to provide a Deep Neural Network library)

您现在需要有NVidia卡才能在GPU上使用Apache MXNet进行开箱即用的训练。有人要求支持AMD,但到目前为止还没有实施https://github.com/apache/incubator-mxnet/issues/621

你可以尝试使用TVM(Github(。这是一个深度学习编译器,可以采用MXNet模型并将其编译为中间表示,可以在多个平台上运行。

请参阅关于如何编译MXNet型号的具体示例

最新更新