How can I use AMD RX Radeon with mxnet package in R specifically

Question

I read about the possibility to use NVidia with mxnet package, however I didn't find anything about using RX Radeon GPU. can I specifically use RX Radeon with mxnet package in R? Below is the link that explains mxnet Installation with NVidia: 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)

Possible duplicate of [OpenCL / AMD: Deep Learning](https://stackoverflow.com/questions/30622805/opencl-amd-deep-learning) — phiver, Jul 11 '18 at 15:28

score 1 · Answer 1 · answered Aug 07 '18 at 20:14

You need to have NVidia card at the moment to train on GPU using Apache MXNet out of the box. There were requests to support AMD, but so far it hasn't been implemented yet - https://github.com/apache/incubator-mxnet/issues/621

You could try and use TVM (Github). This is a deep learning compiler that can take MXNet model and compile it into intermediate representation, that can be run on multiple platforms.

See this specific example on how to compile an MXNet model

How can I use AMD RX Radeon with mxnet package in R specifically

1 Answers1