我如何专门在R中使用带有mxnet软件包的AMD RX Radeon

Question

我了解了将NVidia与mxnet软件包一起使用的可能性，但是我没有发现有关使用RX Radeon GPU的任何信息。我如何在R中将RX Radeon与mxnet包一起使用？ 下面的链接说明了使用NVidia安装mxnet： Installation with 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)

Answer 1

您现在需要立即拥有NVidia卡才能使用Apache MXNet在GPU上进行训练。有人请求支持AMD，但到目前为止尚未实现-https://github.com/apache/incubator-mxnet/issues/621

您可以尝试使用TVM（Github）。这是一个深度学习编译器，可以采用MXNet模型并将其编译为可以在多个平台上运行的中间表示形式。

有关如何编译MXNet模型，请参见此specific example